[JAX] Add an MoE Block (Layer) that compound router, permutation, groupedGEMM and communication#2912
Draft
tdophung wants to merge 28 commits into
Draft
[JAX] Add an MoE Block (Layer) that compound router, permutation, groupedGEMM and communication#2912tdophung wants to merge 28 commits into
tdophung wants to merge 28 commits into
Commits
Commits on May 5, 2026
- committed
- committed
- committed
- committed
- committed
- committed
- committed
![pre-commit-ci[bot]](https://avatars.githubusercontent.com/in/68672?v=4&size=32){kind=avatar}
Commits on May 6, 2026
- committed
Commits on May 12, 2026
- committed
- committed
- committed
- committed
Commits on May 15, 2026
Commits on May 19, 2026
- committed
- committed
- committed
- committed
- committed
- committed
- committed
- committed
- committed