From ae478298f3dc6f611fb968fe2e831e8c9925b243 Mon Sep 17 00:00:00 2001 From: Jake Harmon Date: Wed, 5 Mar 2025 19:24:01 -0800 Subject: [PATCH] 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: 733951719 --- README.md | 4 ++-- dm_pix/_src/augment_test.py | 4 ++-- 2 files changed, 4 insertions(+), 4 deletions(-) diff --git a/README.md b/README.md index bf7766a..b3bfeea 100644 --- a/README.md +++ b/README.md @@ -135,8 +135,8 @@ Please read our [contributing guidelines](./CONTRIBUTING.md) and send us PRs! [Autograd]: https://github.com/hips/autograd "Autograd on GitHub" [DeepMind JAX Ecosystem]: https://deepmind.com/blog/article/using-jax-to-accelerate-our-research "DeepMind JAX Ecosystem" [DeepMind JAX Ecosystem citation]: https://github.com/deepmind/jax/blob/main/deepmind2020jax.txt "Citation" -[JAX]: https://github.com/google/jax "JAX on GitHub" -[JAX installation instructions]: https://github.com/google/jax#installation "JAX installation" +[JAX]: https://github.com/jax-ml/jax "JAX on GitHub" +[JAX installation instructions]: https://github.com/jax-ml/jax#installation "JAX installation" [jit]: https://jax.readthedocs.io/en/latest/jax.html#jax.jit "jax.jit documentation" [NumPy]: https://numpy.org/ "NumPy" [pmap]: https://jax.readthedocs.io/en/latest/jax.html#jax.pmap "jax.pmap documentation" diff --git a/dm_pix/_src/augment_test.py b/dm_pix/_src/augment_test.py index 85a137e..f584a17 100644 --- a/dm_pix/_src/augment_test.py +++ b/dm_pix/_src/augment_test.py @@ -189,8 +189,8 @@ def test_flip(self, images_list): probability=(0., 1.)) # Due to a bug in scipy we cannot test all available modes, refer to these - # issues for more information: https://github.com/google/jax/issues/11097, - # https://github.com/google/jax/issues/11097 + # issues for more information: https://github.com/jax-ml/jax/issues/11097, + # https://github.com/jax-ml/jax/issues/11097 @parameterized.named_parameters( ("in_range_nearest_0", _RAND_FLOATS_IN_RANGE, "nearest", 0), ("in_range_nearest_1", _RAND_FLOATS_IN_RANGE, "nearest", 1),