DeepSpeed - Q4 2026 Roadmap
I maintainer di solito rispondono entro 3 giorni
@delock ci sta già lavorando.
Dal 26/9/2026.
Valutazione
Questa issue non è ancora stata valutata.
Descrizione
This is the roadmap for DeepSpeed Q4 2026. Feedback is welcome — please leave comments on this issue or join the #2026q4-roadmap channel on the DeepSpeed Slack.
Some of the items are the extension of the Q3 roadmap: https://github.com/deepspeedai/DeepSpeed/issues/8104
Tagging all the PoCs (code maintainers) for each workstream, the PoCs can coordinate developer meetings, sort out implementation plan, and are responsible for updating the status.
New feature and enhancement
Mixture-of-Expert (MoE) Training Scaling and Kernel Development
Mixture-of-Expert (MoE) is the mainstream model architecture for frontier AI models. Scaling MoE training requires further improvement on the MoE backend of DeepSpeed, enabling better efficiency, scalability and performance.
- Pipelined Communication & Compute (More Efficient All-to-All Communications and Reduce HBM activation spikes) (PoC: @PKUWZP )
- Block-Sparse Kernels for efficient token dropping (PoC: @PKUWZP )
- Hardware-Aware Memory Optimization & Model Sharding (PoC: @PKUWZP )
- Expert Loading Balance & Hotspot Mitigations (PoC: @PKUWZP )
- Resiliency & Silence Data Corruption Detection (PoC: @PKUWZP )
- AutoEP Performance Optimization & Parallel Folding (AutoEP + AutoTP) (PoC: @tohtana @pengdurice )
- MoE gradient norm and clipping (PoC @pengdurice )
- FP8 GEMM for MoE (PoC @pengdurice )
- Improvements of preset and configuration definitions for better coverage of MoE variants (PoC @pengdurice )
DeepCompile
- Automatic optimization pass orchestration (@tohtana @sfc-gh-truwase @pengdurice )
- Expert parallel support (@tohtana @pengdurice )
- AutoSP with ZeRO (@tohtana )
- Model coverage improvements including VL/VLA models (@tohtana @sfc-gh-truwase , optionally @PKUWZP )
- Save/load optimized IR graphs (@tohtana )
- Robustness improvements (@tohtana )
New Accelerators Support (Q4)
- DeepSpeed TPU accelerators support and performance optimization (@PKUWZP @tohtana @sfc-gh-truwase )
- Apple Silicon Accelerators support for DeepSpeed (@PKUWZP )
- Performance Optimization on Emerging AI Accelerators (@PKUWZP @delock )
DeepSpeed HybridEngine Optimization (Q4)
- Achieving on-par performance with small batch size compared to mainstream LLM Inference Engines (@delock @PKUWZP )
- Continuous Batching Rollout support (@delock )
- Kernel Injection for co-located rollout support (@delock )
Enable training of large scale models
- Enable autoEP, autoTP, Ulysses under Z3 to unblock large scale model training ( POC @pengdurice @tohtana )
- Test and verify support of SOTA models (model specific kernels, attention implementations can go to DeepSpeedExamples) ( POC @pengdurice )
- Muon optimizer offload( POC @delock @pengdurice )
On-Policy Distillation (OPD) Trainer
- Systems Optimization for Large model OPD across multi-accelerators. (@delock @minjiazhang @PKUWZP )
- Coupled with HybridEngine Optimization for end-to-end benchmarking (@delock @minjiazhang @PKUWZP )
- Documentation for OPD trainer onboarding (@delock @PKUWZP )
Code quality and health (Optional)
- Continue deprecating legacy codes (PoC @sfc-gh-truwase )
- Refactor the current code base, i.e. unified process group placement, unified optimizer initialization, reduce usages of globals, fix inconsistent code and doc descriptions. (PoC @sfc-gh-truwase @pengdurice )
AI Agent-driven Distributed Training Optimization (Optional)
- DeepSpeed-native Agentic training optimization (TBD, tentatively @PKUWZP )
Agent Sandbox enablement for Agentic post-training capabilities (Optional)
- Implement Agent Sandbox Environment for DeepSpeed to enable Agentic Post-training (@PKUWZP )
Universal Checkpointing (UCP) feature extension (Optional)
- Affine mapping as Semantic free UCP info (Phase 1) (@delock )
- UCP info support Yuan model TP sharding. (@delock )
- Lingua principale
- Python
- Stelle
- 43.3k
- Fork
- 5.1k
- Merge medio
- 4g 21h
- PR unite (30g)
- 64
Preparare l'ambiente
- Nessun Dockerfile né file Docker Compose
- Nessun modello di pull request
- Leggi la guida per i contributori
Come iniziare
- Leggi tutta la issue e poi la guida ai contributi del progetto.
- Commenta sulla issue per dire che te ne occupi tu — evita che due persone facciano lo stesso lavoro.
- Fai un fork del repository e lavora su un branch.
- Apri una pull request che faccia riferimento al numero della issue.
Altre issue di deepspeedai/DeepSpeed
-
Difficoltà 2/5 1-3 ore Idoneità per principianti 76/100
deepspeedai/DeepSpeed#8925 · 1 commento ·
I maintainer di solito rispondono entro 3 giorni
-
[BUG] Universal checkpoint resume puts CPU `step` into fused/capturable optimizer stateForse già presa @ArjunPakhan l’ha presa 6 giorni fa. Aperta
Difficoltà 2/5 1-3 ore Idoneità per principianti 74/100
deepspeedai/DeepSpeed#8884 · 1 commento ·
I maintainer di solito rispondono entro 3 giorni
-
init_inference rejects dtype=torch.int8 on every acceleratorForse di nuovo libera Una pull request per questa issue è stata chiusa senza essere unita. Aperta
Difficoltà 2/5 1-3 ore Idoneità per principianti 78/100
deepspeedai/DeepSpeed#8577 ·
I maintainer di solito rispondono entro 3 giorni
-
[BUG] AutoTP checkpoint load raises AttributeError on a model containing nn.InstanceNorm1dForse già presa @vineethsaivs l’ha presa 44 giorni fa. Aperta
Difficoltà 2/5 1-3 ore Idoneità per principianti 84/100
deepspeedai/DeepSpeed#8514 ·
I maintainer di solito rispondono entro 3 giorni
-
[BUG] Python checkpoint writers fail to reload full ZeRO-3 state with default torch.loadForse già presa @ai-yang l’ha presa 45 giorni fa. Aperta
Difficoltà 2/5 1-3 ore Idoneità per principianti 78/100
deepspeedai/DeepSpeed#8500 ·
I maintainer di solito rispondono entro 3 giorni
Tutte le issue di deepspeedai/DeepSpeed
Issue simili
-
Difficoltà 1/5 Meno di un'ora Idoneità per principianti 72/100
521xueweihan/HelloGitHub#4150 ·
-
Difficoltà 2/5 1-3 ore Idoneità per principianti 74/100
aws-samples/sample-per-model-bedrock#595 ·
I maintainer di solito rispondono entro 1 giorno
-
Difficoltà 1/5 Meno di un'ora Idoneità per principianti 85/100
I maintainer di solito rispondono entro 1 giorno
-
bug connectors
Difficoltà 1/5 Meno di un'ora Idoneità per principianti 67/100
pyinfra-dev/pyinfra#2036 ·
I maintainer di solito rispondono entro 9 giorni
-
Difficoltà 2/5 1-3 ore Idoneità per principianti 72/100
I maintainer di solito rispondono entro 1 giorno