Commit
This commit does not belong to any branch on this repository, and may belong to a fork outside of the repository.
Update references to JAX's GitHub repo
JAX has moved from https://github.com/google/jax to https://github.com/jax-ml/jax PiperOrigin-RevId: 702886838
- Loading branch information