Skip to content

[JAX] Compare the attention tutorial to its baseline on a shorter sequence - #3828

Merged
KshitijLakhani merged 4 commits into
NVIDIA:mainfrom
aybchan:fix/jax-attention-tutorial-test-memory
Nov 7, 2026
Merged

[JAX] Compare the attention tutorial to its baseline on a shorter sequence#3828
KshitijLakhani merged 4 commits into
NVIDIA:mainfrom
aybchan:fix/jax-attention-tutorial-test-memory

[JAX] Generalize the attention tutorial test comments

df78bf7
Select commit
Loading
Failed to load commit list.
DCO / DCO succeeded Nov 6, 2026 in 1s

DCO

All commits are signed off!

View more details on DCO