jax
5c614470 - [Pallas TPU] Add lowerings for scalar `absf` and `rsqrt`

Commit

1 year ago

[Pallas TPU] Add lowerings for scalar `absf` and `rsqrt` This PR is similar to https://github.com/jax-ml/jax/pull/24284 PiperOrigin-RevId: 689546724

References

Author

ayaka14732

Committer

Google-ML-Automation

Parents