jax

5f5e7423 - Mark as thread-unsafe tests that modify possibly-cached jaxprs in-place.