jax
c60bafcc - [Pallas TPU] Fix lowering for `jnp.remainder`

Commit

1 year ago

[Pallas TPU] Fix lowering for `jnp.remainder` Fixes https://github.com/jax-ml/jax/issues/24027 PiperOrigin-RevId: 688614799

References

Author

ayaka14732

Committer

Google-ML-Automation

Parents