deepspeed has 1.4M monthly downloads and 43K GitHub stars. Version 0.19.2, first published in 2020.
It is actively maintained under a permissive license, last released in Jun 2026.
It is built on top of einops, hjson and msgpack. It is used by 99 tracked packages, including protenix, openrlhf and project-llm-trainer.
GitHub topics include deep-learning, pytorch and gpu.
Package Insights
((week_daily_avg - month_daily_avg) / month_daily_avg) * 100Weekly Downloads
GitHub Stars
Downloads by OS
Python Versions
Top Countries
Dependencies
- einops
- hjson
- msgpack
- ninja
- numpy
- packaging >=20.0
- psutil
- py-cpuinfo
- pydantic >=2.0.0
- torch >=2.0.0
- tqdm
91 optional dependencies
- accelerate[all]
- accelerate[dev]
- autodoc-pydantic[all]
- autodoc-pydantic[readthedocs]
- clang-format[all]
- clang-format[dev]
- comet-ml[all]
- comet-ml[dev]
- deepspeed-kernels[all]
- deepspeed-kernels[dev]
- diffusers[sd]
- diffusers[all]
- docutils[readthedocs]
- docutils[dev]
- docutils[all]
- future[dev]
- future[all]
- google[all]
- google[inf]
- hjson[all]
- hjson[readthedocs]
- hjson[autotuning-ml]
- importlib-metadata[dev]
- importlib-metadata[all]
- lm-eval[all]
- lm-eval[inf]
- mpi4py[all]
- mpi4py[1bit-mpi]
- mup[dev]
- mup[all]
- neural-compressor[all]
- neural-compressor[sparse]
- packaging[all]
- packaging[readthedocs]
- pre-commit[all]
- pre-commit[dev]
- protobuf[inf]
- protobuf[all]
- psutil[readthedocs]
- psutil[all]
- py-cpuinfo[all]
- py-cpuinfo[readthedocs]
- pydantic[readthedocs]
- pydantic[all]
- pytest[dev]
- pytest[all]
- pytest-forked[dev]
- pytest-forked[all]
- pytest-randomly[all]
- pytest-randomly[dev]
- pytest-xdist[all]
- pytest-xdist[dev]
- qtorch[inf]
- qtorch[all]
- qtorch[dev]
- recommonmark[readthedocs]
- recommonmark[dev]
- recommonmark[all]
- safetensors[inf]
- safetensors[all]
- scipy[all]
- scipy[deepcompile]
- sentencepiece[inf]
- sentencepiece[all]
- sphinx[dev]
- sphinx[all]
- sphinx-rtd-theme[dev]
- sphinx-rtd-theme[all]
- sphinx-rtd-theme[readthedocs]
- tabulate[all]
- tabulate[autotuning]
- tabulate[autotuning-ml]
- tensorboard[dev]
- tensorboard[all]
- torch[readthedocs]
- torch[all]
- torchvision[dev]
- torchvision[all]
- tqdm[all]
- tqdm[readthedocs]
- transformers[dev]
- transformers[all]
- transformers[inf]
- triton[all]
- triton[sd]
- triton[sparse-attn]
- triton[triton]
- wandb[all]
- wandb[dev]
- xgboost[all]
- xgboost[autotuning-ml]
Used By
- protenix
- openrlhf
- project-llm-trainer
- resemble-enhance
- factory-sdk
- arbor-ai
- arctic-training
- hpsv3
- llm-optimized-inference
- alphafold3-pytorch-lightning-hydra
- opentau
- deepspeed-mii
- trill-proteins
- chunkformer
- gpt3-torch
- charylu-models
- oat-llm
- pyuino
- lean-dojo-v2
- efficientllm-toolkit
- aivoifu
- aistrainer
- arctic-platform
- t2v-metrics
- cosyvoice2-eu
- neotune
- transformers-inference-toolkit
- semviqa
- translation-canvas
- gpt4-torch
- aeiva
- tuningtron
- andromeda-torch
- renate
- small-vlm
- cm3
- cosyvoice2-fr-infer
- omnivector-embed
- unipercept-reward
- resp-agent
- autobiasdetector
- alignment-handbook
- asrchild
- phi-torch
- thebestllmever
- dbgpt-hub
- easy-lightning
- narration-xtts2
- collie-lm
- memorag