ColossalAI/colossalai/kernel
1SAA 219df6e685 Optimized MoE layer and fixed some bugs;
Decreased moe tests;

Added FFNExperts and ViTMoE model
2022-03-11 15:50:28 +08:00
..
cuda_native Optimized MoE layer and fixed some bugs; 2022-03-11 15:50:28 +08:00
jit adapted for sequence parallel (#163) 2022-01-20 13:44:51 +08:00
__init__.py refactor kernel (#142) 2022-01-13 16:47:17 +08:00