jax
f3115d32 - Fix dtype failures in JaxGroupedQueryAttentionReferenceTest.

Commit

305 days ago

Fix dtype failures in JaxGroupedQueryAttentionReferenceTest. PiperOrigin-RevId: 746097962

References

Author

dfm

Committer

Google-ML-Automation

Parents