jax
b6e4b938 - Add jaxlib_extension_version guard against explicit copying

Commit

301 days ago

Add jaxlib_extension_version guard against explicit copying in jax.device_put. PiperOrigin-RevId: 744838237

References

Author

pschuh

Committer

Google-ML-Automation

Parents