MCPcopy Create free account
hub / github.com/deepspeedai/DeepSpeed

github.com/deepspeedai/DeepSpeed

Chat with this repo
repository ↗ · DeepWiki ↗ · release v0.19.4 ↗ · + Follow · compare 3 versions
13,179 symbols 51,170 edges 1,271 files ⚖ custom 2,387 documented · 18% updated todayv0.19.4 · 2026-08-06★ 42,8711,163 open issues

Browse by type

Functions 11,258 Types & classes 1,887 Endpoints 34
What it actually does AI analysis from the code graph — generated when you open this
loading…
README

License Apache 2.0 PyPI version Downloads Build OpenSSF Best Practices Twitter Japanese Twitter Chinese Zhihu Slack

Office Hours

DeepSpeed hosts regular office hours on the last Tuesday of each month at 12:00 America/New_York to discuss development plans, features, etc. This meeting is public for anyone to join and ask questions. The meeting is hosted on Zoom and can be joined here.

Latest News

More news


Extreme Speed and Scale for DL Training

DeepSpeed enabled the world's most powerful language models (at the time of this writing) such as MT-530B and BLOOM. DeepSpeed offers a confluence of system innovations, that has made large scale DL training effective, and efficient, greatly improved ease of use, and redefined the DL training landscape in terms of scale that is possible. These innovations include ZeRO, ZeRO-Infinity, 3D-Parallelism, Ulysses Sequence Parallelism, DeepSpeed-MoE, etc.


DeepSpeed Adoption

DeepSpeed was an important part of Microsoft’s AI at Scale initiative to enable next-generation AI capabilities at scale, where you can find more information here.

DeepSpeed has been used to train many different large-scale models, below is a list of several examples that we are aware of (if you'd like to include your model please submit a PR):

DeepSpeed has been integrated with several different popular open-source DL frameworks such as:

Documentation
Transformers with DeepSpeed
Accelerate with DeepSpeed
Lightning with DeepSpeed
MosaicML with DeepSpeed
Determined with DeepSpeed
MMEngine with DeepSpeed

Build Pipeline Status

Description Status
NVIDIA nv-pre-compile-ops modal-torch-latest
AMD amd-mi200
CPU torch-latest-cpu
Intel Gaudi hpu-gaudi2
Intel XPU xpu-max1100
Integrations aws-accelerate
Misc Formatting pages-build-deployment Documentation Statuspython
Huawei Ascend NPU Huawei Ascend NPU

Installation

The quickest way to get started with DeepSpeed is via pip, this will install the latest release of DeepSpeed which is not tied to specific PyTorch or CUDA versions. DeepSpeed includes several C++/CUDA extensions that we commonly refer to as our 'ops'. By default, all of these extensions/ops will be built just-in-time (JIT) using torch's JIT C++ extension loader that relies on ninja to build and dynamically link them at runtime.

Requirements

  • PyTorch must be installed before installing DeepSpeed.
  • For full feature support we recommend a version of PyTorch that is >= 2.0 and ideally the latest PyTorch stable release.
  • A CUDA or ROCm compiler such as nvcc or hipcc used to compile C++/CUDA/HIP extensions.
  • Specific GPUs we develop and test against are listed below, this doesn't mean your GPU will not work if it doesn't fall into this category it's just DeepSpeed is most well tested on the following:
  • NVIDIA: Pascal, Volta, Ampere, and Hopper architectures
  • AMD: MI100 and MI200

Contributed HW support

  • DeepSpeed now support various HW accelerators.
Contributor Hardware Accelerator Name Contributor validated Upstream validated
Huawei Huawei Ascend NPU npu Yes No
Intel Intel(R) Gaudi(R) 2 AI accelerator hpu Yes Yes
Intel Intel(R) Xeon(R) Processors cpu Yes Yes
Intel Intel(R) Data Center GPU Max series xpu Yes Yes
Tecorigin Scalable Data Analytics Accelerator sdaa Yes No

PyPI

We regularly push releases to PyPI and encourage users to install from there in most cases.

pip install deepspeed

After installation, you can validate your install and see which extensions/ops your machine is compatible with via the DeepSpeed environment report.

ds_report

If you would l

Core symbols most depended-on inside this repo

browse all functions →

Shape

Method 7,608
Function 3,650
Class 1,871
Route 34
Enum 16

Languages

Python89%
C++11%

Modules by API surface

deepspeed/runtime/engine.py336 symbols
deepspeed/runtime/zero/stage3.py183 symbols
deepspeed/runtime/zero/stage_1_and_2.py148 symbols
csrc/deepspeed4science/evoformer_attn/iterators/predicated_tile_iterator_residual_last.h133 symbols
deepspeed/runtime/zero/partition_parameters.py127 symbols
deepspeed/module_inject/layers.py117 symbols
tests/unit/v1/zero/test_zero_user_backward.py113 symbols
tests/unit/v1/zero/test_zero.py113 symbols
csrc/deepspeed4science/evoformer_attn/iterators/predicated_tile_access_iterator_residual_last.h111 symbols
tests/unit/v1/moe/test_autoep_unit.py96 symbols
deepspeed/runtime/utils.py84 symbols
deepspeed/runtime/config.py78 symbols

Dependencies from manifests, versioned

autodoc_pydantic2.0.0 · 1×
clang-format18.1.3 · 1×
comet_ml3.41.0 · 1×
diffusers0.25.0 · 1×
importlib-metadata4 · 1×
lm-eval0.3.0 · 1×
neural-compressor2.1.0 · 1×
packaging20.0 · 1×
pre-commit3.2.0 · 1×
pydantic2.0.0 · 1×
pytest7.2.0 · 1×
qtorch0.3.0 · 1×

For agents

$ claude mcp add DeepSpeed \
  -- python -m otcore.mcp_server <graph>

⬇ download graph artifact

Ask about this repo answers extend the page