onnxruntime/orttraining
pengwa a4180d79c5
Multi-tensor SGDOptimizer (on device training) (#14083)
Implement SGDOptimizerV2 taking sequence of weights and gradients as
inputs.

For CPU EP and CUDA EP only.

Added tests.
2023-01-11 10:15:53 -08:00
..
orttraining Multi-tensor SGDOptimizer (on device training) (#14083) 2023-01-11 10:15:53 -08:00
pytorch_frontend_examples Set black's target version (#11370) 2022-04-27 14:52:19 -07:00
tools Refactor training build options (#13964) 2023-01-03 13:28:16 -08:00