jax
a24420e7 - [Pallas TPU] Add lowering for `lax.cos_p`

Commit

1 year ago

[Pallas TPU] Add lowering for `lax.cos_p` Fixes https://github.com/jax-ml/jax/issues/24026 PiperOrigin-RevId: 680754948

Author

ayaka14732

Committer

Google-ML-Automation

Parents