Skip to content

[JAX] Allow input tensors to be sharded with TP in EP dispatch/combine ops - #3283

Draft
phu0ngng wants to merge 1 commit into
NVIDIA:mainfrom
phu0ngng:jax_etp
Draft

[JAX] Allow input tensors to be sharded with TP in EP dispatch/combine ops#3283
phu0ngng wants to merge 1 commit into
NVIDIA:mainfrom
phu0ngng:jax_etp

Commits

Commits on Jul 30, 2026