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