Skip to content

Releases: deepspeedai/DeepSpeed

v0.19.4 Patch Release

Choose a tag to compare

@loadams loadams released this 06 Aug 02:10
c455031

What's Changed

  • Validate warmup_type in WarmupCosineLR like WarmupLR by @sohumt123 in #8151
  • Run pull request code only inside a Modal Sandbox by @tohtana in #8170
  • [AutoTP] Allow ZeRO stage 3 inference with tensor parallelism by @delock in #8167
  • Fix autotuning get_val_by_key to search all nested subdicts by @ebarkhordar in #8177
  • Enable support for Tutel when k != 1 for shared moe by @jinyouzhi in #8174
  • NVMe write warning by @sfc-gh-truwase in #8164
  • Guard LRRangeTest and OneCycle schedulers against zero step sizes by @ebarkhordar in #8166
  • Fix WarmupLR collapsing multi-group base LRs to group 0's by @ebarkhordar in #8171
  • Stabilize fork-sensitive nightly tests and AutoSP coverage by @tohtana in #8172
  • Make the OneCycle stair counts actually change the schedule by @vineethsaivs in #8179
  • [AutoTP] Enable HF colwise_gather_output to support lm_head replace by @jinyouzhi in #8146
  • Scope DeepCompile compiler state to graph and engine lifecycles by @tohtana in #8159
  • Clarify merge commit exemption for signoff check in AGENTS.md and CLAUDE.md by @jinyouzhi in #8194
  • [AutoTP] Preserve universal checkpoint metadata for HuggingFace tp_plan by @jinyouzhi in #8187
  • Derive AutoEP rank splits from the per-expert count exchange by @yh0903 in #8190
  • Warn when zero.Init silently falls back to a single rank (#8084) by @akshansh47 in #8089
  • Release GIL in blocking AIO bindings by @tohtana in #8196
  • Enable ZeRO-3 linear wrapper for existing models by @tohtana in #8189
  • Skip CUDA transformer forward tests when TransformerBuilder is incompatible by @sfc-gh-truwase in #8200
  • Clamp capacity to num_tokens in MoE gating drop branches (complete #5353) by @ebarkhordar in #8155
  • Preserve tensor learning rates across scheduler updates by @n33levo in #8202
  • [AutoTP] Fix ZeRO-3 checkpoint consolidation to gather across TP and DP by @delock in #8168
  • Add managed_gradient_accumulation for ZeRO stage 0/1 by @sfc-gh-truwase in #8184
  • MoE dispatch: fix silent Tutel + TP corruption, speed up native and Tutel paths by @jinyouzhi in #8195
  • Add keyword argument support to activation checkpointing by @zupengwang in #8182
  • Remove redundant DCO workflow by @tohtana in #8188
  • Preserve tensor learning rates in OneCycle by @tohtana in #8205
  • Add Triton grouped-GEMM for MoE experts on Ampere/Ada by @hwchen2017 in #8180
  • Wiring up offload_opt_states by @pengdurice in #8186
  • Fix DeepCompile ZeRO-3 gathered parameter ownership by @tohtana in #8157
  • Consolidate LR scheduler tensor-LR tests into a contract suite by @delock in #8206
  • Fix MLU_Accelerator conformance to the DeepSpeedAccelerator ABC by @ebarkhordar in #8208
  • Reduce redundant work in AutoEP token routing by @hwchen2017 in #8209
  • Make PipelineModule.set_checkpoint_interval actually change the interval by @vineethsaivs in #8178
  • Share DeepNVMe pinned-tensor manager and route swap buffers through I/O handles by @sfc-gh-truwase in #8212
  • Stop trim_mean sorting the caller's list in place by @vineethsaivs in #8199
  • Unmanaged gradient accumulation: ZeRO stage 2 support by @sfc-gh-truwase in #8203
  • Fix ZeRO-3 all_reduce param fetch stride for padded parameters by @ebarkhordar in #8158
  • Unmanaged gradient accumulation: ZeRO stage 3 support by @sfc-gh-truwase in #8217
  • Update version.txt pre-0.19.4 release by @loadams in #8218

New Contributors

Full Changelog: v0.19.3...v0.19.4

v0.19.3 Patch Release

Choose a tag to compare

@loadams loadams released this 23 Jul 01:08
0c36f6d

What's Changed

  • Update version post release by @loadams in #8070
  • Validate fp16 dynamic loss scaling parameters are positive by @aryanputta in #8050
  • Add merge queue DCO workflow by @tohtana in #8086
  • Stop obsolete CI jobs on workflow cancellation by @tohtana in #8034
  • Fix ZeRO-3: Use per-param dtype for output buffers in _allgather_params_coalesced by @albertvillanova in #8073
  • fix: close file descriptor in deepspeed_io_handle_t::wait() to prevent fd leak by @MarkCLChang in #8075
  • Add type hints to top-level public API functions by @arunshar in #8079
  • Default gradient_clipping to 1.0 by @sfc-gh-truwase in #8068
  • activation_checkpointing: default num_layers to None so configure() assert fires by @Kymi808 in #8041
  • Add configurable engine log level by @sfc-gh-truwase in #8067
  • [CI] diff driven test selection by @stas00 in #8077
  • Enable bf16 check_grad_overflow by default (matching fp16) by @yongzhe-wang in #8035
  • Support AutoEP with ZeRO-3 zero.Init source modules by @tohtana in #8060
  • Update publication list in README.md by @PKUWZP in #8095
  • fix: AutoTP partition_config uses full hierarchical module path by @delock in #8088
  • Reject Muon optimizer with reduce_scatter in ZeRO-1/2 by @whycoming in #8090
  • Fix DeepCompile profile metadata backfill by @tohtana in #8094
  • Add numerical-correctness test for Muon under ZeRO-1/2 by @whycoming in #8091
  • Don't swallow KeyboardInterrupt/SystemExit in NPUOpBuilder by @ajinkyajawale14499 in #8096
  • Avoid CUDA context initialization during op compatibility checks at import by @Achyuthan-S in #8078
  • [DeepCompile] fix gather params in dynamo skipped frames for ZeRO3 by @XAheli in #8059
  • Feat: zero3 deprecate elastic checkpoint by @nathon-lee in #8099
  • Fix eigenvalue parsing for compression-only quantize configs by @sowndappan5 in #8057
  • fix: log eigenvalue monitor values by @he-yufeng in #8049
  • Fix DeepCompile profiling memory cleanup by @tohtana in #8106
  • Add optional torchembed RoPE backend to apply_rotary_pos_emb by @py-ai-dev in #8052
  • Fix minor comment/docstring typos in runtime and inference modules by @nathon-lee in #8046
  • feat(zenflow): run the overlapped CPU optimizer in a native process by @Antlera in #8058
  • fix: add buffer-length check in shm.cpp by @orbisai0security in #8082
  • Fix FlopsProfiler crash when dp_world_size is None under sequence parallelism by @vineethsaivs in #8122
  • Fix Python 3.9 import-time TypeError in AutoEP ep_router by @vineethsaivs in #8119
  • Add input validation for save_checkpoint and warmup LR schedulers by @jahnavi-yelamanchi in #8126
  • Fix ZeRO-3: use per-param dtype for quantize-scale buffer in _allgather_params_coalesced by @roycho96 in #8128
  • Fix ZeRO-3 autocast gather with mixed parameter dtypes by @tohtana in #8113
  • doc: Remove suggestion to build extensions in parallel by @Flamefire in #7899
  • feat(types): add type hints to public comm API functions by @NovusEdge in #8118
  • Add Hybrid Engine rollout in DeepSpeed to support On-Policy Distillation (OPSD) Trainer by @PKUWZP in #8027
  • Add AutoEP + AutoTP parallel folding by @tohtana in #8064
  • Make DCO workflow Probot compatible by @tohtana in #8110
  • ZeRO 1/2: wait on all IPG-bucket producer streams in average_tensor (#8061) by @arunshar in #8080
  • Fix CommsLogger.stop_profiling_comms not disabling global profiling by @vineethsaivs in #8137
  • Guard WarmupCosineLR against total_num_steps == warmup_num_steps (ZeroDivisionError) by @vineethsaivs in #8142
  • Add EXAONE 4.5 model support for Inference V2 by @Bias92 in #8121
  • fix(autotp): when using autotp, ignore the consistency of certain data within the tp_group. by @1787648106 in #8125
  • Reduce blockDim in fake_quantize_kernel for improved SM occupancy by @flutist in #8115
  • Fix DeepSpeedInferenceConfig crash on bool moe backward-compat value by @winklemad in #8145
  • fix(exceptions): narrow broad Exception clauses to specific types by @sowndappan5 in #8136
  • fix(muon): support ZeRO-1/2 reduce scatter by @Micuks in #8141
  • [DeepCompile] Add lightweight pass contracts for optimization passes by @jahnavi-yelamanchi in #8139
  • Fix DeepCompile fallback handling for scalar outputs by @tohtana in #8149
  • Use synthetic CIFAR data in AlexNet tests by @tohtana in #8143
  • Update CI badge by @tohtana in #8150
  • Update Zhipeng Wang affiliation to Google in COMMITTERS.md by @PKUWZP in #8153
  • Fix ZeRO-3 hooks for attribute-delegating modules by @Functionhx in #8144
  • Fix repeated gradient scaling across pipeline stages by @Functionhx in #8154
  • Z3: Support for activation ckpt with frozen params by @sfc-gh-truwase in #8148
  • Adopt PTF Code of Conduct by @sfc-gh-truwase in #8165

New Contributors

Full Changelog: v0.19.2...v0.19.3

v0.19.2 Patch Release

Choose a tag to compare

@loadams loadams released this 16 Jun 20:52
b919284

What's Changed

  • fix(fp16): filter requires_grad in FP16 optimizer flat buffer init by @avicooper1 in #8029
  • Run AutoSP compile tests sequentially by @tohtana in #8020
  • Fix PR-target workflow concurrency groups by @tohtana in #8017
  • Fix full CI test isolation for ZeRO chmod and NVMe quantization tests by @tohtana in #8008
  • Keep required CI checks visible for ignored paths by @tohtana in #8019
  • Bump version by @sfc-gh-truwase in #8030
  • Add engine.coalesce_grad_reduction() for ZeRO 1/2/3 multi-backward by @roycho96 in #7992
  • feat(zero): enable torch.func transforms on engine for ZeRO 0/1/2 by @roycho96 in #8026
  • Simplify module_inject.transpose by @xbcReal in #8028
  • Fix DeepCompile all-gather scheduler candidate selection by @tohtana in #8033
  • Version fix to unblock pypi by @sfc-gh-truwase in #8039
  • Bump version after 0.19.1 release by @tohtana in #8040
  • Fix DeepCompile ZeRO-3 release parameter lifetime by @tohtana in #8032
  • Fix ZenFlow ZeRO-3 selective optimizer crash with parameter offload on nvme by @Antlera in #8042
  • Add test coverage for Muon muon_lr/adam_lr overrides by @sowndappan5 in #8047
  • Avoid HF Hub access in CPU unit test setup by @tohtana in #8053
  • Fix DeepCompile ZeRO-1 grad target lifetime by @tohtana in #8036
  • Normalize ZeRO-3 DeepCompile grad dtype before reduction by @tohtana in #8038
  • Remove AutoSP assertion against Transformers version by @tohtana in #8044
  • fix(transformer): use correct stride in Transpose_Kernel shared memory indexing to eliminate bank conflicts by @flutist in #8055
  • zero3: invalidate coordinator trace on hook re-registration by @roycho96 in #8043
  • Consistent fp32 grads flow by @sfc-gh-truwase in #8056
  • Add AutoEP by @tohtana in #7938
  • Fix: ZenFlow Adam integration for updated PyTorch backward flow (#7759) by @Antlera in #7771
  • Pass expected grad dtype to register_z3_param in ZeRO-3 release test by @tohtana in #8063
  • Add Biren SUPA accelerator support by @frozenleaves in #8054
  • Mixed-precision: per-policy param/buffer dtype cast (preserve fp32 buffers) by @sfc-gh-truwase in #8066

New Contributors

Full Changelog: v0.19.1...v0.19.2

v0.19.1

Choose a tag to compare

@sfc-gh-truwase sfc-gh-truwase released this 27 May 13:19
510ebe5

What's Changed

New Contributors

Full Changelog: v0.19.0...v0.19.1

v0.19.0

Choose a tag to compare

@loadams loadams released this 06 May 21:47
98a2fbd

What's Changed

New Contributors

Full Changelog: v0.18.9...v0.19.0

v0.18.9 Patch Release

Choose a tag to compare

@loadams loadams released this 30 Mar 16:42
8c93851

What's Changed

  • Respect $TRITON_HOME by @Flamefire in #7907
  • Add Feature Universal Checkpoint for AutoTP by @nathon-lee in #7908
  • fix: remove unnecessary shell=True in ROCm GPU architecture detection by @instantraaamen in #7915
  • Don't detect local GPU if $DS_IGNORE_CUDA_DETECTION is set by @Flamefire in #7896
  • Add HuggingFace tp_plan support for AutoTP by @delock in #7901
  • fix: handle non-existent path in is_nfs_path for Triton autotune cache by @Krishnachaitanyakc in #7921
  • Fix backward compatibility of torch.amp.custom_fwd for PyTorch < 2.4 by @tohtana in #7920
  • Extending Muon Optimizer Support for ZeRO Stage 3 by @PKUWZP in #7919
  • Add news item for ASPLOS 2026 Best Paper Award by @PKUWZP in #7923
  • fix(superoffload) preserve multi-group updates with shared cpu buffers (#7905) by @xylian86 in #7906
  • AGENTS.md: Add pre-commit command to existing CI requirements line by @delock in #7930
  • Update README with latest news from DeepSpeed by @PKUWZP in #7931
  • Merging AutoSP into DeepSpeed by @neeldani in #7860
  • Add fallback to full test by @tohtana in #7933
  • Remove Microsoft Corporation copyright from AGENTS.md and CLAUDE.md by @PKUWZP in #7932
  • Update version.txt for latest incoming release 0.18.9 by @loadams in #7935

New Contributors

Full Changelog: v0.18.8...v0.18.9

v0.18.8 Patch Release

Choose a tag to compare

@loadams loadams released this 13 Mar 18:48
5f7b687

What's Changed

New Contributors

Full Changelog: v0.18.7...v0.18.8

v0.18.7 Patch Release

Choose a tag to compare

@loadams loadams released this 05 Mar 20:43
6c59d54

What's Changed

New Contributors

Full Changelog: v0.18.6...v0.18.7

v0.18.6 Patch Release

Choose a tag to compare

@loadams loadams released this 12 Feb 23:21
d2ca6e7

What's Changed

New Contributors

Full Changelog: v0.18.5...v0.18.6

v0.18.5 Patch Release

Choose a tag to compare

@loadams loadams released this 30 Jan 18:21
b19987c

What's Changed

New Contributors

Full Changelog: v0.18.4...v0.18.5