[JAX] EP Dispatch with overflow detection option - #3277

Open

phu0ngng wants to merge 1 commit into

NVIDIA:mainfrom

phu0ngng:ep-jax-overflow

Commits on Jul 29, 2026

[JAX] Add NCCL EP drop-on-overflow policy and total_recv_tokens output
phu0ngng
committed