jax
a66f606f - [Pallas TPU] Roll back exposing accumulator references to be used with explicit MXU APIs.

Commit

77 days ago

[Pallas TPU] Roll back exposing accumulator references to be used with explicit MXU APIs. Reverts effd7d771d33f5f7ab7bfdd49230ce3c82362515 PiperOrigin-RevId: 965633135

References

#40027 - [Pallas TPU] Roll back exposing accumulator references to be used with explicit MXU APIs.

Author

bchetioui

Committer

Google-ML-Automation

Parents

2014e4c5

jax a66f606f - [Pallas TPU] Roll back exposing accumulator references to be used with explicit MXU APIs.

jax
a66f606f - [Pallas TPU] Roll back exposing accumulator references to be used with explicit MXU APIs.