ColossalAI/colossalai/nn
Xuanlei Zhao dd2c28a323
[npu] use extension for op builder (#5172)
* update extension

* update cpu adam

* update is

* add doc for cpu adam

* update kernel

* update commit

* update flash

* update memory efficient

* update flash attn

* update flash attention loader

* update api

* fix

* update doc

* update example time limit

* reverse change

* fix doc

* remove useless kernel

* fix

* not use warning

* update

* update
2024-01-08 11:39:16 +08:00
..
layer [moe] merge moe into main (#4978) 2023-11-02 02:21:24 +00:00
loss [moe] merge moe into main (#4978) 2023-11-02 02:21:24 +00:00
lr_scheduler [hotfix] fix lr scheduler bug in torch 2.0 (#4864) 2023-10-12 14:04:24 +08:00
optimizer [npu] use extension for op builder (#5172) 2024-01-08 11:39:16 +08:00
__init__.py [legacy] move communication and nn to legacy and refactor logger (#4671) 2023-09-11 16:24:28 +08:00
init.py [misc] update pre-commit and run all files (#4752) 2023-09-19 14:20:26 +08:00