jax
ecd37959 - [Mosaic TPU] Allow more input types in the low-level MXU APIs

Commit

25 days ago

[Mosaic TPU] Allow more input types in the low-level MXU APIs PiperOrigin-RevId: 859052030

References

Author

apaszke

Committer

Google-ML-Automation

Parents