jax
695ee8f3 - Fix a race in pjit under free threading.

Commit

330 days ago

Fix a race in pjit under free threading. Fixes https://github.com/jax-ml/jax/issues/27767

References

Author

hawkinsp

Parents