1
0
mirror of https://github.com/hpcaitech/ColossalAI.git synced 2025-05-06 15:38:26 +00:00
ColossalAI/colossalai/legacy/nn
Frank Lee 7cfed5f076
[feat] refactored extension module ()
* [feat] refactored extension module

* polish

* polish

* polish

* polish

* polish

* polish

* polish

* polish

* polish

* polish
2024-01-25 17:01:48 +08:00
..
_ops [misc] update pre-commit and run all files () 2023-09-19 14:20:26 +08:00
layer [feat] refactored extension module () 2024-01-25 17:01:48 +08:00
loss [npu] change device to accelerator api () 2024-01-09 10:20:05 +08:00
metric [misc] update pre-commit and run all files () 2023-09-19 14:20:26 +08:00
parallel [misc] update pre-commit and run all files () 2023-09-19 14:20:26 +08:00
__init__.py [legacy] clean up legacy code () 2023-09-18 16:31:06 +08:00