Olatunji Ruwase 7f90ef4bdd Multiple zero stage 3 related fixes (#3886) 1 year ago
..
activation_checkpointing b361c72761 Update DeepSpeed copyright license to Apache 2.0 (#3111) 1 year ago
comm b361c72761 Update DeepSpeed copyright license to Apache 2.0 (#3111) 1 year ago
half_precision 6b2365e4fa Re-enable elastic training for torch 2+ (#4010) 1 year ago
pipe aef6c65ce3 Reduce Unit Test Times (Part 3) (#3850) 1 year ago
sparse_tensor b361c72761 Update DeepSpeed copyright license to Apache 2.0 (#3111) 1 year ago
utils b361c72761 Update DeepSpeed copyright license to Apache 2.0 (#3111) 1 year ago
zero 7f90ef4bdd Multiple zero stage 3 related fixes (#3886) 1 year ago
test_autocast.py b361c72761 Update DeepSpeed copyright license to Apache 2.0 (#3111) 1 year ago
test_data.py aef6c65ce3 Reduce Unit Test Times (Part 3) (#3850) 1 year ago
test_data_efficiency.py b361c72761 Update DeepSpeed copyright license to Apache 2.0 (#3111) 1 year ago
test_ds_config_dict.py b361c72761 Update DeepSpeed copyright license to Apache 2.0 (#3111) 1 year ago
test_ds_config_model.py b361c72761 Update DeepSpeed copyright license to Apache 2.0 (#3111) 1 year ago
test_ds_initialize.py 6b2365e4fa Re-enable elastic training for torch 2+ (#4010) 1 year ago
test_lr_schedulers.py b361c72761 Update DeepSpeed copyright license to Apache 2.0 (#3111) 1 year ago
test_multi_output_model.py b361c72761 Update DeepSpeed copyright license to Apache 2.0 (#3111) 1 year ago
test_pld.py b361c72761 Update DeepSpeed copyright license to Apache 2.0 (#3111) 1 year ago
test_runtime_utils.py b361c72761 Update DeepSpeed copyright license to Apache 2.0 (#3111) 1 year ago