onnxruntime/orttraining/orttraining/python/training
2024-10-15 16:47:16 -07:00
..
amp
api
experimental
onnxblock Fix training artifacts for 2GB+ models and MSELoss (#22414) 2024-10-15 16:47:16 -07:00
optim Fix typos - 1st Wave (#21278) 2024-07-11 13:35:08 +08:00
ort_triton Update ruff and clang-format versions (#21479) 2024-07-24 11:50:11 -07:00
ortmodule Fix Orttraining Linux Lazy Tensor CI Pipeline (#21652) 2024-08-21 18:10:08 +08:00
utils ORTModule GraphTransitionManager (#19007) 2024-07-03 10:53:31 +08:00
__init__.py Bump ruff to 0.3.2 and black to 24 (#19878) 2024-03-13 10:00:32 -07:00
_utils.py
artifacts.py Enable >2GB models + allow model paths to be passed for generate_artifacts API (#20958) 2024-06-21 09:55:26 -07:00