onnxruntime/orttraining/orttraining/python/training/optim
..
__init__.py
_apex_amp_modifier.py
_ds_code_store.py
_ds_modifier.py
_megatron_modifier.py
_modifier.py
_modifier_registry.py
_multi_tensor_apply.py
config.py
fp16_optimizer.py FP16 optimizer automatically detect DeepSpeed compatibility (#18084) 2023-10-25 15:11:02 +08:00
fused_adam.py
lr_scheduler.py