jax
b550aade - Merge pull request #38486 from Cjkkkk:flex_attn

Commit

42 days ago

Merge pull request #38486 from Cjkkkk:flex_attn PiperOrigin-RevId: 933552461

References

#38486 - Reenable cuDNN SDPA flex attention with version guard >= 9.13 to avoid NaNs

Author

Google-ML-Automation

Google-ML-Automation

Parents

Loading