Adam Louly fb4707f76d add cuda support to python bindings (#13700) ... ### Description Add cuda support to the on device training python bindings. ### Motivation and Context Now users can set the execution provider (cpu or cuda) when using python bindings for on device training apis.		2022-12-08 16:03:53 -08:00
..
amp	Format all python files under onnxruntime with black and isort (#11324)	2022-04-26 09:35:16 -07:00
api	add cuda support to python bindings (#13700)	2022-12-08 16:03:53 -08:00
experimental	Make ORT callable from various Pytorch compilers (LazyTensor, TorchDynamo, etc) (#10460)	2022-08-22 09:40:40 -07:00
onnxblock	Fix on-device training ExportModelForInferencing api (#13510)	2022-10-31 21:29:06 -07:00
optim	[ORTModule] Update Supported DeepSpeed Version for FP16_Optimizer (#13305)	2022-10-13 13:03:01 +08:00
ortmodule	Move a function call outside a loop in ORTModule (#13771)	2022-11-30 12:49:41 +01:00
torchdynamo	Improve DORT document (#13790)	2022-11-30 16:55:25 -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