jax
4efa56d2 - Merge pull request #28872 from Cjkkkk:jax_cudnn_sdpa_mla

Commit

1 year ago

Merge pull request #28872 from Cjkkkk:jax_cudnn_sdpa_mla PiperOrigin-RevId: 773127984

References

Author

Google-ML-Automation

Parents