jax
bc33c63d - [jax] Explicitly declare ones as jnp.float32

Commit

121 days ago

[jax] Explicitly declare ones as jnp.float32 PiperOrigin-RevId: 817316970

References

#32486 - [jax] Explicitly declare ones as jnp.float32

Author

basioli-k

basioli-k

Committer

Google-ML-Automation

Google-ML-Automation

Parents

Loading