Change references in setup.py and utilities to reference the JAX repo…

… move to the JAX-ML org PiperOrigin-RevId: 675265804
jax-ml · Sep 16, 2024 · 010bef8 · 010bef8
1 parent 9408606
commit 010bef8
Show file tree

Hide file tree

Showing 4 changed files with 6 additions and 6 deletions.
diff --git a/.github/workflows/self_hosted_runner_utils/setup_runner.sh b/.github/workflows/self_hosted_runner_utils/setup_runner.sh
@@ -31,7 +31,7 @@ runner_token="$3"
 # - sets empty string as default to avoid unbound variable error from set -u
 jax_repo_url="${4-}"
 if [ -z "${jax_repo_url}" ]; then
-  jax_repo_url="https://github.com/google/jax"
+  jax_repo_url="https://github.com/jax-ml/jax"
 fi
 
 # Create `runner` user. This user won't have sudo access unless you ssh into the
@@ -67,7 +67,7 @@ cd ~/
 
 git clone ${jax_repo_url}
 
-# Based on https://github.com/google/jax/settings/actions/runners/new
+# Based on https://github.com/jax-ml/jax/settings/actions/runners/new
 # (will be 404 for github users with insufficient repo permissions)
 mkdir actions-runner && cd actions-runner
 curl -o actions-runner-linux-x64.tar.gz -L ${actions_runner_download}

diff --git a/jax_plugins/cuda/plugin_setup.py b/jax_plugins/cuda/plugin_setup.py
@@ -66,13 +66,13 @@ def has_ext_modules(self):
           # dependency via, for example, cuSOLVER. NVIDIA's cuSOLVER packages
           # do not have a version constraint on their dependencies, so the
           # package doesn't get upgraded even though not doing that can cause
-          # problems (https://github.com/google/jax/issues/18027#issuecomment-1756305196)
+          # problems (https://github.com/jax-ml/jax/issues/18027#issuecomment-1756305196)
           # Until NVIDIA add version constraints, add a version constraint
           # here.
           "nvidia-nvjitlink-cu12>=12.1.105",
       ],
     },
-    url="https://github.com/google/jax",
+    url="https://github.com/jax-ml/jax",
     license="Apache-2.0",
     classifiers=[
         "Development Status :: 3 - Alpha",

diff --git a/jax_plugins/cuda/setup.py b/jax_plugins/cuda/setup.py
@@ -48,7 +48,7 @@ def load_version_module(pkg_path):
     author_email="jax-dev@google.com",
     packages=packages,
     install_requires=[],
-    url="https://github.com/google/jax",
+    url="https://github.com/jax-ml/jax",
     license="Apache-2.0",
     classifiers=[
         "Development Status :: 3 - Alpha",

diff --git a/jax_plugins/rocm/setup.py b/jax_plugins/rocm/setup.py
@@ -48,7 +48,7 @@ def load_version_module(pkg_path):
     author_email="Ruturaj.Vaidya@amd.com",
     packages=packages,
     install_requires=[],
-    url="https://github.com/google/jax",
+    url="https://github.com/jax-ml/jax",
     license="Apache-2.0",
     classifiers=[
         "Development Status :: 3 - Alpha",