jax
52ae9e22 - [Mosaic] Use BF16 ops for math::PowF on TPUv6+.

Commit

323 days ago

[Mosaic] Use BF16 ops for math::PowF on TPUv6+. PiperOrigin-RevId: 772444344

References

#476 - CI: 06/19/25 upstream sync

#594 - Add CI workflow for Pytest ROCm

Author

WindQAQ

WindQAQ

Committer

charleshofer

charleshofer

Parents

Loading