Skip to content

[JAX] Expert Parallelism: JAX primitives + VJPs#3036

Merged
phu0ngng merged 24 commits into
NVIDIA:mainfrom
phu0ngng:phuong/ep-3-jax
Jun 27, 2026
Merged

[JAX] Expert Parallelism: JAX primitives + VJPs#3036
phu0ngng merged 24 commits into
NVIDIA:mainfrom
phu0ngng:phuong/ep-3-jax

Merge branch 'main' into phuong/ep-3-jax

9187d81
Select commit
Loading
Failed to load commit list.
DCO / DCO succeeded Jun 27, 2026 in 1s

DCO

All commits are signed off!

View more details on DCO