onnxruntime/orttraining/orttraining/python/training/ortmodule
..
experimental
graph_optimizers
torch_cpp_extensions
__init__.py
_custom_autograd_function.py
_custom_autograd_function_exporter.py
_custom_autograd_function_runner.py
_custom_gradient_registry.py
_custom_op_symbolic_registry.py
_execution_agent.py
_fallback.py
_fallback_exceptions.py
_gradient_accumulation_manager.py
_graph_execution_interface.py
_graph_execution_manager.py
_graph_execution_manager_factory.py
_inference_manager.py
_io.py
_logger.py
_onnx_models.py
_runtime_inspector.py
_torch_module_factory.py
_torch_module_interface.py
_torch_module_ort.py
_torch_module_pytorch.py
_training_manager.py
_utils.py
_zero_stage3_compatibility.py
graph_optimizer_registry.py
options.py
ortmodule.py