ColossalAI/tests/test_optimizer
Frank Lee 7cfed5f076
[feat] refactored extension module (#5298)
* [feat] refactored extension module

* polish

* polish

* polish

* polish

* polish

* polish

* polish

* polish

* polish

* polish
2024-01-25 17:01:48 +08:00
..
test_adam_kernel.py [feat] refactored extension module (#5298) 2024-01-25 17:01:48 +08:00
test_adam_optim.py
test_nvme.py [test] merge old components to test to model zoo (#4945) 2023-10-20 10:35:08 +08:00