jax
de8fa8fd - [Mosaic TPU] Add support for sqrt and rsqrt in bf16 on TPUv6

Commit

1 year ago

[Mosaic TPU] Add support for sqrt and rsqrt in bf16 on TPUv6 PiperOrigin-RevId: 708016513

References

Author

apaszke

Committer

Google-ML-Automation

Parents