This website requires JavaScript.
Explore
Help
Register
Sign In
github
/
ColossalAI
Watch
1
Star
0
Fork
0
You've already forked ColossalAI
mirror of
https://github.com/hpcaitech/ColossalAI.git
synced
2026-04-12 15:14:55 +00:00
Code
Issues
Packages
Projects
Releases
Wiki
Activity
Files
6185b9772d41f9caa5776a48173aff2fe13bf829
ColossalAI
/
colossalai
/
kernel
/
cuda_native
History
1SAA
e7a1136436
Optimized MoE layer and fixed some bugs;
...
Decreased moe tests; Added FFNExperts and ViTMoE model
2022-02-22 10:27:12 +08:00
..
csrc
Optimized MoE layer and fixed some bugs;
2022-02-22 10:27:12 +08:00
__init__.py
refactor kernel (
#142
)
2022-01-13 16:47:17 +08:00
layer_norm.py
moved env variables to global variables; (
#215
)
2022-02-15 11:31:13 +08:00
multihead_attention.py
Fixed docstring in colossalai (
#171
)
2022-01-21 10:44:30 +08:00
scaled_softmax.py
Fixed docstring in colossalai (
#171
)
2022-01-21 10:44:30 +08:00