littsk 83b52c56cd [feature] Add clip_grad_norm for hybrid_parallel_plugin (#4837) ...

* Add clip_grad_norm for hibrid_parallel_plugin

* polish code

* add unittests

* Move tp to a higher-level optimizer interface.

* bug fix

* polish code

2023-10-12 11:32:37 +08:00

..

naive_amp

[feature] Add clip_grad_norm for hybrid_parallel_plugin (#4837)

2023-10-12 11:32:37 +08:00

__init__.py

[legacy] clean up legacy code (#4743)

2023-09-18 16:31:06 +08:00