Official account for DeepSpeed, a library that enables unprecedented scale and speed for deep learning training + inference. 日本語 : @DeepSpeedAI_JP

Filter
Exclude
Time range
-
Minimum likes
Come and listening to all the exciting updates from @DeepSpeedAI and talk to our awesome @PKUWZP
🎤 Meet keynote speaker Zhipeng Wang (@PKUWZP), Senior Staff Software Engineer at @Google and DeepSpeed maintainer. bit.ly/4iVl3mo At #PyTorchCon North America, he’ll share advances helping DeepSpeed train large models efficiently across diverse hardware. October 20-21 | San Jose Register: bit.ly/4sh3DSw
3
435
DeepSpeed Office Hours: Tue Aug 25, 12pm ET v0.19.3–0.19.5: • Grad accum for custom & serverless (Tinker) • AutoEP+AutoTP folding & ZeRO-3 TP inference • Hybrid Engine for on-policy distillation • Pinned-memory CPU/NVMe offload speedups Join: zoom-lfx.platform.linuxfound…
2
3
1,226
Replying to @ezyang @StasBekman
Edward, thanks for the write up. My offloading experience aligns with the key points. However, the doc affirms cuda-specific, and user control limitations. I agree that current torch works for most use cases.
2
31
Replying to @StasBekman @ezyang
Edward, I found some documentation that suggests deleting the tensor will force unpinning. Is that correct? We would also prefer to use PyTorch API directly. Thanks!
1
2
88
The meeting link for tomorrow's DeepSpeed developer meetings is now available: Aug. 7, 9:00 a.m. PT — DeepCompile efficiency & robustness Aug. 7, 10:00 a.m. PT — Expert Parallelism enhancements Everyone interested is welcome to join! zoom-lfx.platform.linuxfound…
Apologies for the short notice! We're holding two DeepSpeed developer meetings on Aug. 7 to discuss our Q3 roadmap goals: 9:00 a.m. PT — DeepCompile efficiency & robustness 10:00 a.m. PT — Expert Parallelism enhancements Roadmap: github.com/deepspeedai/DeepS… Join Slack: deepspeedworkspace.slack.com…
3
922
It is neat to see how to preserve vmem mappings across the phy mem release/reclaim. I have a question. For offloading (and reloading), it seems you would need to use enable_cpu_backup mode github.com/fzyzcjy/torch_mem…
1
2
71
Replying to @ezyang @StasBekman
The limitations we encountered include - Does not work on CPU-only - No unpin() API instead uses GC, and so less user control.
1
3
292
Apologies for the short notice! We're holding two DeepSpeed developer meetings on Aug. 7 to discuss our Q3 roadmap goals: 9:00 a.m. PT — DeepCompile efficiency & robustness 10:00 a.m. PT — Expert Parallelism enhancements Roadmap: github.com/deepspeedai/DeepS… Join Slack: deepspeedworkspace.slack.com…
8
1,627
We will host our monthly Office Hour for DeepSpeed users on July 28th, 9AM PT. Feel free to join and discuss any topics ranging from issues to roadmaps. Zoom link: zoom-lfx.platform.linuxfound…
3
819
We now have native support for all ZeRO stages 1/2/3 for Muon Optimizers, providing superior performance on LLM pre-training and post-training. Feel free to try it out, kudos to @PKUWZP Guokai Ma, Peng Du and Chi for the contribution!
DeepSpeed now supports the Muon Optimizer. Optimized specifically for internal 2D weights within neural networks, Muon is gaining traction for its significant memory savings and strong convergence metrics during LLM training. In our latest blog post, the DeepSpeed team shares a deep dive into their integration setup, implementation of hybrid optimizer strategies, and early benchmark results. @PKUWZP Read the full technical breakdown here 👉 bit.ly/4dLjGE2
3
13
1,692
Reminder to join our DeepSpeed Office Hours on Tuesday, May 26 at 12:00 PM America/New_York. We'll cover general questions, Q2 roadmap progress, and requests for Q3. Everyone is welcome! Zoom: zoom-lfx.platform.linuxfound…
3
299
Great News! Thanks to DeepSpeed AutoSP, efficient long context LLM training is now easily accessible.
Want to train LLMs on longer contexts without re-engineering your entire systems stack? Introducing AutoSP — the first compiler-based solution that automatically optimizes LLM training for long contexts. Under the hood, AutoSP applies a series of compiler passes that trigger sequence parallelism, paired with a curated activation-checkpointing scheme tailored for long-context training. It's integrated directly into DeepSpeed, so enabling long-context training is just a config change away. No more rewiring your stack to push context lengths. Read the blog to learn more 🖇️ pytorch.org/blog/introducing… ✍ @AhanGupta13, Zhihao W., Neel Dani, @toh_tana, Tunji Ruwase, @_Minjia_Zhang_ #PyTorch #DeepSpeed #AutoSP #OpenSourceAI
1
2
4
1,492
Training Optimization on Multimodal models is an important pillar for pushing the frontier Multimodal Foundation Model development. Kudos to @toh_tana and Tunji Ruwase for their excellent work. It's just the starting point, more to come!
New @DeepSpeedAI updates make large-scale multimodal training simpler and more memory-efficient. Our latest blog introduces a PyTorch-identical backward API that helps code multimodal training loops easy, plus low-precision model states (BF16/FP16) that can reduce peak memory by up to 40% when combined with torch.autocast. 🖇️ Read the full post for details: hubs.la/Q044yYVs0 #DeepSpeed #PyTorch #MemoryEfficiency #MultimodalTraining #OpenSourceAI
4
5
1,237
It's nice to share the most recent updates from the DeepSpeed project at #PyTorchCon, we will continue pushing the boundary of LLM distributed training for the OSS community.
🎙️ Mic check: Tunji Ruwase, Lead, DeepSpeed Project & Principal Engineer at Snowflake, is bringing the 🔥 to the keynote stage at #PyTorchCon! Get ready for big ideas and deeper learning October 22–23 in San Francisco. 👀 Speakers: hubs.la/Q03GPYFn0 🎟️ hubs.la/Q03GPXVH0
1
7
1,305
UIUC, AnyScale, and Snowflake significantly enhanced LLM offloading for the Superchip era!
🚀 SuperOffload: Unleashing the Power of Large-Scale LLM Training on Superchips Superchips like the NVIDIA GH200 offer tightly coupled GPU-CPU architectures for AI workloads. But most existing offloading techniques were designed for traditional PCIe-based systems. Are we truly tapping into their full potential for LLM training? 🎯 SuperOffload is our answer to this challenge, a new DeepSpeed component rethinking offloading from the ground up, specially designed for LLM training on Superchips. ✨ SuperOffload is exact -- no approximation, no heuristics, and no changes to your training algorithm. Just faster, larger model with longer sequence training using the same code, which are made possible by system-level optimizations exploiting Superchip architecture. 🧪 SuperOffload allows you: - Finetune models like GPT-OSS-20B, Qwen3-14B, and Phi-4 on a single GH200 - Up to 4X faster speed than previous approaches like ZeRO-Offload - Effortlessly scales to: -- Qwen3-30B-A3B and Seed-OSS-36B on 2 x GH200s -- LLaMA2-70B on 4 x GH200s -- 1M sequence length on 8x GH200 with 55% MFU - Easy-to-use: Fully integrated and open-sourced in DeepSpeed. Just a few lines of code to enable! 📚 Read more through official PyTorch blog: pytorch.org/blog/superoffloa… 🧠 For more technical details, please read our technical report: arxiv.org/abs/2509.21271 🛠️ SuperOffload is fully open-sourced through DeepSpeed. Try it now: github.com/deepspeedai/DeepS… 📄 SuperOffload has been accepted to ASPLOS 2026! Kudos to Xinyu Lian (@Alexlian0806), Masahiro Tanaka (@toh_tana), and Olatunji Ruwase. 🎤 Featured at PyTorch Conference 2025 SuperOffload will be featured in the DeepSpeed & vLLM keynote at this year's PyTorch Conference in San Francisco. 🔥Come see how we're rethinking large-scale LLM training for the Superchip era: events.linuxfoundation.org/p…
3
12
2,726
Step into the future of AI at #PyTorchCon 2025, Oct 22–23 in San Francisco 🔥 Join the DeepSpeed keynote and technical talks. Register: events.linuxfoundation.org/p… + Oct 21 co-located events: Measuring Intelligence, Open Agent & AI Infra Summits / Startup Showcase & PyTorch Training
2
7
2,903
ZenFlow is a massive improvement to DeepSpeed Offloading. Courtesy of an excellent collaboration among University of Virginia, UC Merced, Argonne National Laboratory, Microsoft, and Snowflake.
Introducing #ZenFlow: No Compromising Speed for #LLM Training w/ Offloading 5× faster LLM training with offloading 85% less GPU stalls 2× lower I/O overhead 🚀 Blog: hubs.la/Q03DJ6GJ0 🚀 Try ZenFlow and experience 5× faster training with offloading: hubs.la/Q03DJ6Vb0
10
1,798