name: deepspeed version: 0.19.7 summary: DeepSpeed library license: Apache Software License 2.0 author: DeepSpeed Team homepage: http://deepspeed.ai install_size: 1.8 MB module_format: sdist-only doc_url: https://deepspeed.readthedocs.io repo_url: https://github.com/deepspeedai/DeepSpeed last_released_at: 2026-09-16T22:01:06Z releases_last_12mo: 18 avg_days_between_releases: 16.5 ## dependencies einops hjson msgpack ninja numpy packaging >=20.0 psutil py-cpuinfo pydantic >=2.0.0 torch >=2.0.0 tqdm ## extras [1bit-mpi] mpi4py [all] py-cpuinfo torchembed >=0.3.1 importlib-metadata >=4 mpi4py autodoc_pydantic >=2.0.0 neural-compressor ==2.1.0 clang-format ==18.1.3 tabulate qtorch packaging mup sphinx-rtd-theme xgboost transformers >=4.32.1 transformers >=4.51.3 scipy triton >=2.1.0 sentencepiece torch >=2.0.0 docutils <0.18 tensorboard pytest <8.4.0,>=7.2.0 hjson diffusers >=0.25.0 sphinx pytest-forked deepspeed-kernels pytest-randomly tqdm future safetensors psutil pydantic >=2.0.0 pre-commit >=3.2.0 comet_ml >=3.41.0 protobuf qtorch ==0.3.0 recommonmark torchvision sphinx_rtd_theme wandb google accelerate pytest-xdist lm-eval ==0.3.0 [autotuning] tabulate [autotuning-ml] hjson tabulate xgboost [deepcompile] scipy [dev] accelerate clang-format ==18.1.3 comet_ml >=3.41.0 deepspeed-kernels docutils <0.18 future importlib-metadata >=4 mup pre-commit >=3.2.0 pytest <8.4.0,>=7.2.0 pytest-forked pytest-randomly pytest-xdist qtorch ==0.3.0 recommonmark sphinx sphinx-rtd-theme tensorboard torchvision transformers >=4.51.3 wandb [inf] google lm-eval ==0.3.0 protobuf qtorch safetensors sentencepiece transformers >=4.32.1 [readthedocs] autodoc_pydantic >=2.0.0 docutils <0.18 hjson packaging psutil py-cpuinfo pydantic >=2.0.0 recommonmark sphinx_rtd_theme torch >=2.0.0 tqdm [sd] diffusers >=0.25.0 triton >=2.1.0 [sparse] neural-compressor ==2.1.0 [torchembed] torchembed >=0.3.1 ## maintainers DeepSpeed Team