Skip to content

[JAX] Support calling MOE router kernels from JAX side#2711

Open
tdophung wants to merge 10 commits intoNVIDIA:mainfrom
tdophung:router_jax
Open

[JAX] Support calling MOE router kernels from JAX side#2711
tdophung wants to merge 10 commits intoNVIDIA:mainfrom
tdophung:router_jax

Commits

Commits on Feb 25, 2026

Commits on Feb 26, 2026

Commits on Feb 28, 2026