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

Commit

209 days ago

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

References

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

Author

WindQAQ

WindQAQ

Committer

charleshofer

charleshofer

Parents

Loading