jax
b1760ad6 - [Pallas:TPU] Add a new low-level MXU API for TPUv7x

Commit

23 days ago

[Pallas:TPU] Add a new low-level MXU API for TPUv7x PiperOrigin-RevId: 859017335

References

#34463 - [Pallas:TPU] Add a new low-level MXU API for TPUv7x

Author

apaszke

apaszke

Committer

Google-ML-Automation

Google-ML-Automation

Parents

Loading