jax
ea53c761 - Fix thread safety of JAX error checking

Commit

1 year ago

Fix thread safety of JAX error checking Fix thread safety of JAX error checking by making the global states thread local PiperOrigin-RevId: 733164878

References

Author

ayaka14732

Committer

Google-ML-Automation

Parents