chenlin/Mujoco_WASM
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Update references to JAX's GitHub repo

Browse Source 
JAX has moved from https://github.com/google/jax to https://github.com/jax-ml/jax

PiperOrigin-RevId: 733951719
Change-Id: Iee4a7bf4bcc9fc3108ca9d30b6b98f5df20c79c7
This commit is contained in:
Jake Harmon
2025-03-05 19:24:01 -08:00
committed by Copybara-Service
parent 0f563ecf31
commit a0fc6b18b5
2 changed files with 2 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files
mjx/mujoco/mjx/_src/math.py
+1 -1
View File
@@ -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