Matthew Hoffman 604d701e35 Introduce pydantic_v1 compatibility module for pydantic>=2.0.0 support (#4407) 1 年之前
..
activation_checkpointing 42c1e916f6 feat(activation_checkpointing): add `non_reentrant_checkpoint` to support inputs require no grad (#4118) 1 年之前
comm b361c72761 Update DeepSpeed copyright license to Apache 2.0 (#3111) 1 年之前
half_precision 57a27b0803 add type checker ignore to resolve that pylance can't resolved noqa annotation (#4102) 1 年之前
pipe 60a3e89eed use ```non_reentrant_checkpoint``` fix requires_grad of input must be true for activation checkpoint layer in pipeline train. (#4224) 1 年之前
sparse_tensor b361c72761 Update DeepSpeed copyright license to Apache 2.0 (#3111) 1 年之前
utils b361c72761 Update DeepSpeed copyright license to Apache 2.0 (#3111) 1 年之前
zero 9adc73ff65 Handle empty parameter groups (#4277) 1 年之前
test_autocast.py b361c72761 Update DeepSpeed copyright license to Apache 2.0 (#3111) 1 年之前
test_data.py aef6c65ce3 Reduce Unit Test Times (Part 3) (#3850) 1 年之前
test_data_efficiency.py b361c72761 Update DeepSpeed copyright license to Apache 2.0 (#3111) 1 年之前
test_ds_config_dict.py b361c72761 Update DeepSpeed copyright license to Apache 2.0 (#3111) 1 年之前
test_ds_config_model.py 604d701e35 Introduce pydantic_v1 compatibility module for pydantic>=2.0.0 support (#4407) 1 年之前
test_ds_initialize.py 6b2365e4fa Re-enable elastic training for torch 2+ (#4010) 1 年之前
test_lr_schedulers.py b361c72761 Update DeepSpeed copyright license to Apache 2.0 (#3111) 1 年之前
test_multi_output_model.py b361c72761 Update DeepSpeed copyright license to Apache 2.0 (#3111) 1 年之前
test_mup_optimizers.py 9647ea791d Add MuP optimizers (#2043) 1 年之前
test_pld.py b361c72761 Update DeepSpeed copyright license to Apache 2.0 (#3111) 1 年之前
test_runtime_utils.py b361c72761 Update DeepSpeed copyright license to Apache 2.0 (#3111) 1 年之前