Skip to content

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

Draft
phu0ngng wants to merge 2 commits 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 2 commits into
NVIDIA:mainfrom
phu0ngng:jax_etp

[pre-commit.ci] auto fixes from pre-commit.com hooks

7ff6022
Select commit
Loading
Failed to load commit list.
DCO / DCO succeeded Oct 20, 2026 in 1s

DCO

All commits are signed off!

View more details on DCO