Skip to content

[JAX] Return max_logit from TE fused attn - #3112

Open
KshitijLakhani wants to merge 22 commits into
NVIDIA:mainfrom
KshitijLakhani:klakhani/feat/return-softmax-stats
Open

[JAX] Return max_logit from TE fused attn#3112
KshitijLakhani wants to merge 22 commits into
NVIDIA:mainfrom
KshitijLakhani:klakhani/feat/return-softmax-stats

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

d7bf4cf
Select commit
Loading
Failed to load commit list.
DCO / DCO succeeded Aug 24, 2026 in 1s

DCO

All commits are signed off!

View more details on DCO