jax
1f6152d1 - [Pallas] Use Pallas cost estimator for flash attention.

Commit

1 year ago

[Pallas] Use Pallas cost estimator for flash attention. PiperOrigin-RevId: 698573265

References

Author

justinjfu

Committer

Google-ML-Automation

Parents