From a0fc6b18b56de127032d920a1aa16c9edd9072f4 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 Change-Id: Iee4a7bf4bcc9fc3108ca9d30b6b98f5df20c79c7 --- mjx/README.md | 2 +- mjx/mujoco/mjx/_src/math.py | 2 +- 2 files changed, 2 insertions(+), 2 deletions(-) diff --git a/mjx/README.md b/mjx/README.md index 4bb727bd..f4796a5f 100644 --- a/mjx/README.md +++ b/mjx/README.md @@ -9,7 +9,7 @@ This package is a re-implementation of the [MuJoCo physics engine](https://github.com/google-deepmind/mujoco) in -[JAX](https://github.com/google/jax). This library is developed and maintained +[JAX](https://github.com/jax-ml/jax). This library is developed and maintained by Google DeepMind, and is kept up-to-date with the latest developments in MuJoCo itself. diff --git a/mjx/mujoco/mjx/_src/math.py b/mjx/mujoco/mjx/_src/math.py index c7fffbcf..bd3b6019 100644 --- a/mjx/mujoco/mjx/_src/math.py +++ b/mjx/mujoco/mjx/_src/math.py @@ -51,7 +51,7 @@ def norm( """Calculates a linalg.norm(x) that's safe for gradients at x=0. Avoids a poorly defined gradient for jnp.linal.norm(0) see - https://github.com/google/jax/issues/3058 for details + https://github.com/jax-ml/jax/issues/3058 for details Args: x: A jnp.array axis: The axis along which to compute the norm