jax
fe577b5d - [Pallas/Mosaic GPU] Enable `ops_test` for Mosaic GPU.

Commit

354 days ago

[Pallas/Mosaic GPU] Enable `ops_test` for Mosaic GPU. For now, most of the tests are skipped. PiperOrigin-RevId: 734026728

References

Author

bchetioui

Committer

Google-ML-Automation

Parents