[JAX] Schedule EP dispatch/combine on XLA collective stream - #3231

Merged

phu0ngng merged 5 commits into

NVIDIA:mainfrom

phu0ngng:ep/jax_col_stream

Jul 22, 2026

Commits on Jul 21, 2026

[JAX] Schedule EP dispatch/combine on XLA collective stream
phu0ngng
committed
d4428a7
Update tests/jax/test_multi_process_ep.py

phu0ngng
and
greptile-apps[bot]
authored
5d3271e
[JAX] Gate EP collective-stream annotation on JAX/XLA version
phu0ngng
committed
2b62e3c

Commits on Jul 22, 2026

[JAX] Silence pylint not-callable on compute_on EP annotation
phu0ngng
committed
9371934
Merge branch 'main' into ep/jax_col_stream
phu0ngng
authored
7ddd50b