jax
362fb7ae - Remove code to support jaxlib < 0.5.3.

Commit

1 year ago

Remove code to support jaxlib < 0.5.3. The new xla_extension_version is 320. PiperOrigin-RevId: 738522486

References

Author

hawkinsp

Committer

Google-ML-Automation

Parents