jax
48205f93 - [Pallas:TPU] Add a two MXU test for the low-level MXU API

Commit

14 days ago

[Pallas:TPU] Add a two MXU test for the low-level MXU API PiperOrigin-RevId: 859069650

References

Author

apaszke

Committer

Google-ML-Automation

Parents