Skip to content

[JAX] Migrate EP collective-stream annotation to region-based compute_on - #3304

Merged
phu0ngng merged 4 commits into
NVIDIA:mainfrom
phu0ngng:jax_compute_on_version
Aug 6, 2026
Merged

[JAX] Migrate EP collective-stream annotation to region-based compute_on#3304
phu0ngng merged 4 commits into
NVIDIA:mainfrom
phu0ngng:jax_compute_on_version

Merge branch 'main' into jax_compute_on_version

5467b2c
Select commit
Loading
Failed to load commit list.
DCO / DCO succeeded Aug 6, 2026 in 1s

DCO

All commits are signed off!

View more details on DCO