jax
1edb94ec - [XLA][Mosaic] Add support for fp8 matmuls in TPUv5+

Commit

1 year ago

[XLA][Mosaic] Add support for fp8 matmuls in TPUv5+ Needed a little more backfill for TPU load PiperOrigin-RevId: 639206243

References

Author

Google-ML-Automation

Committer

a-googler

Parents