ColossalAI/extensions/optimizer at 1d626233ce8dbf35405cb7d92a5638ee1d830e8f - ColossalAI - Gitea: Git with a cup of tea at home

github/ColossalAI

mirror of https://github.com/hpcaitech/ColossalAI.git synced 2025-12-25 05:22:43 +00:00

Files

History

xs_courtesy 095c070a6e refactor code

2024-03-11 17:06:57 +08:00

..

__init__.py

[feat] refactored extension module (#5298 )

2024-01-25 17:01:48 +08:00

fused_optimizer_cuda.py

refactor code

2024-03-11 17:06:57 +08:00