Wei-Sheng Chin a0a5f57581 [DORT] Use new FX-to-ONNX exporter (#16450) ... The ONNX exporter in DORT have been moved to PyTorch as a formal feature. We therefore switch to consume the exporter from PyTorch instead of maintaining two duplicates.		2023-07-04 13:13:04 -07:00
..
amp	Adopt linrtunner as the linting tool - take 2 (#15085)	2023-03-24 15:29:03 -07:00
api	Python documentation for onnxruntime-training (#15765)	2023-05-02 16:58:16 -07:00
experimental	Manage ORTModule configurations consistently (#16396)	2023-06-27 19:19:36 +08:00
onnxblock	Python documentation for onnxruntime-training (#15765)	2023-05-02 16:58:16 -07:00
optim	Adding this set_to_none flag to zero_grad to have signature parity with pytorch Adam (#16375)	2023-06-19 17:27:41 -07:00
ortmodule	Manage ORTModule configurations consistently (#16396)	2023-06-27 19:19:36 +08:00
torchdynamo	[DORT] Use new FX-to-ONNX exporter (#16450)	2023-07-04 13:13:04 -07:00
utils	Introduce memory observer for ORTModule (#16213)	2023-06-15 15:45:36 +08:00
__init__.py	Refining the offline tooling for training artifact generation (#15212)	2023-03-30 18:05:51 -07:00
_checkpoint_storage.py	Enable pylint and numpy rules (#15218)	2023-03-27 20:37:53 -07:00
_utils.py	Introduce float 8 types (#14731)	2023-05-30 13:25:58 -07:00
artifacts.py	Add ability to create ort format models from training offline utility (#16360)	2023-06-21 18:51:43 -07:00
checkpoint.py	Bump ruff in CI (#15533)	2023-04-17 10:11:44 -07:00
model_desc_validation.py	Adopt linrtunner as the linting tool - take 2 (#15085)	2023-03-24 15:29:03 -07:00
orttrainer.py	Adopt linrtunner as the linting tool - take 2 (#15085)	2023-03-24 15:29:03 -07:00
orttrainer_options.py	Adopt linrtunner as the linting tool - take 2 (#15085)	2023-03-24 15:29:03 -07:00
postprocess.py	Adopt linrtunner as the linting tool - take 2 (#15085)	2023-03-24 15:29:03 -07:00