Skip to content

[JAX] Attention support for explicit sharding - #3446

Draft
jberchtold-nvidia wants to merge 3 commits into
NVIDIA:mainfrom
jberchtold-nvidia:jberchtold/te-attn-explicit-sharding-test
Draft

[JAX] Attention support for explicit sharding#3446
jberchtold-nvidia wants to merge 3 commits into
NVIDIA:mainfrom
jberchtold-nvidia:jberchtold/te-attn-explicit-sharding-test

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

f2b9896
Select commit
Loading
Failed to load commit list.
DCO / DCO succeeded Aug 31, 2026 in 0s

DCO

All commits are signed off!

View more details on DCO