onnxruntime/orttraining/orttraining/python/training
Abhishek Jindal 6fa4555a06
Including support for Deepspeed 0.8.0 (#14506)
### Description
Including Support for Deepspeed 0.8.0.



### Motivation and Context
Deepspeed 0.8.0 has a bug fix and mlfow integration.
2023-02-01 06:19:41 -08:00
..
amp Format all python files under onnxruntime with black and isort (#11324) 2022-04-26 09:35:16 -07:00
api Refactor training build options (#13964) 2023-01-03 13:28:16 -08:00
experimental Make ORT callable from various Pytorch compilers (LazyTensor, TorchDynamo, etc) (#10460) 2022-08-22 09:40:40 -07:00
onnxblock Refactor training build options (#13964) 2023-01-03 13:28:16 -08:00
optim Including support for Deepspeed 0.8.0 (#14506) 2023-02-01 06:19:41 -08:00
ortmodule Enable a single build with optimized inference and on device training (#14241) 2023-01-12 21:36:43 -08:00
torchdynamo Extend ort_backend.py for another ep (#14349) 2023-01-20 07:30:00 -08:00
utils/data Format all python files under onnxruntime with black and isort (#11324) 2022-04-26 09:35:16 -07:00
__init__.py [python] [orttraining] Add utility to export a graph to compute gradients (#8125) 2022-02-18 14:00:49 -08:00
_checkpoint_storage.py Format all python files under onnxruntime with black and isort (#11324) 2022-04-26 09:35:16 -07:00
_utils.py replace 'master' branch ref to 'main' for onnx repo (#12678) 2022-08-30 13:41:42 -07:00
checkpoint.py Format all python files under onnxruntime with black and isort (#11324) 2022-04-26 09:35:16 -07:00
model_desc_validation.py Format all python files under onnxruntime with black and isort (#11324) 2022-04-26 09:35:16 -07:00
orttrainer.py Deprecate ORTTrainer (#13022) 2022-09-23 18:10:09 -07:00
orttrainer_options.py Update ORTModule Default Opset Version to 15 (#12419) 2022-08-05 16:55:04 +08:00
postprocess.py Format all python files under onnxruntime with black and isort (#11324) 2022-04-26 09:35:16 -07:00