jax
4718121e - Merge pull request #25754 from andportnoy:patch-4

Commit

1 year ago

Merge pull request #25754 from andportnoy:patch-4 PiperOrigin-RevId: 713222111

References

#25754 - [Mosaic GPU] Use `num_q_heads=2` in `flash_attention.py`

#196 - CI: 01/08/25 upstream sync

Author

Google-ML-Automation

Google-ML-Automation

Parents

Loading