jax
fa71d7a8 - [Pallas TPU] Expose number of accumulators and number of MXUs for the various TPU

Commit

25 days ago

[Pallas TPU] Expose number of accumulators and number of MXUs for the various TPU generations. PiperOrigin-RevId: 911141237

References

Author

bchetioui

Committer

Google-ML-Automation

Parents