Rename from jax_ml_stack to jax-ai-stack

jax-ml · Sep 12, 2024 · f5da905 · f5da905
1 parent 90c51b4
commit f5da905
Show file tree

Hide file tree

Showing 9 changed files with 10 additions and 10 deletions.
diff --git a/.github/workflows/test.yaml b/.github/workflows/test.yaml
@@ -37,7 +37,7 @@ jobs:
         pip install -U jax flax grain ml_dtypes optax orbax tensorflow tensorflow_datasets pytest pytest-xdist
     - name: Run tests
       run: |
-        pytest -n auto jax_ml_stack
+        pytest -n auto jax_ai_stack
 
   build:
     name: Pinned packages (${{ matrix.os }} Python ${{ matrix.python-version }})
@@ -61,4 +61,4 @@ jobs:
         pip install .[dev,tfds,grain]
     - name: Run tests
       run: |
-        pytest -n auto jax_ml_stack
+        pytest -n auto jax_ai_stack
diff --git a/README.md b/README.md
@@ -1,4 +1,4 @@
-# jax-ml-stack
+# jax-ai-stack
 
 This is a work-in-progress meta-package for the AI/ML stack built on top of the
 [jax](http://github.com/google/jax/) package. It is intended as a location for

diff --git a/jax_ml_stack/__init__.py → jax_ai_stack/__init__.py b/jax_ml_stack/__init__.py → jax_ai_stack/__init__.py
@@ -1,3 +1,3 @@
-"""jax_ml_stack metapackage."""
+"""jax-ai-stack metapackage."""
 
 __version__ = "0.0.0"  # keep in sync with pyproject.toml
diff --git a/jax_ml_stack/tests/__init__.py → jax_ai_stack/tests/__init__.py b/jax_ml_stack/tests/__init__.py → jax_ai_stack/tests/__init__.py
diff --git a/jax_ml_stack/tests/test_nnx_with_grain.py → jax_ai_stack/tests/test_nnx_with_grain.py b/jax_ml_stack/tests/test_nnx_with_grain.py → jax_ai_stack/tests/test_nnx_with_grain.py
diff --git a/jax_ml_stack/tests/test_nnx_with_optax.py → jax_ai_stack/tests/test_nnx_with_optax.py b/jax_ml_stack/tests/test_nnx_with_optax.py → jax_ai_stack/tests/test_nnx_with_optax.py
diff --git a/jax_ml_stack/tests/test_nnx_with_orbax.py → jax_ai_stack/tests/test_nnx_with_orbax.py b/jax_ml_stack/tests/test_nnx_with_orbax.py → jax_ai_stack/tests/test_nnx_with_orbax.py
diff --git a/jax_ml_stack/tests/test_nnx_with_tfds.py → jax_ai_stack/tests/test_nnx_with_tfds.py b/jax_ml_stack/tests/test_nnx_with_tfds.py → jax_ai_stack/tests/test_nnx_with_tfds.py
diff --git a/pyproject.toml b/pyproject.toml
@@ -1,10 +1,10 @@
 [project]
-name = "jax_ml_stack"
-version = "0.0.0"  # keep in sync with jax_ml_stack/__init__.py
+name = "jax-ai-stack"
+version = "0.0.0"  # keep in sync with jax_ai_stack/__init__.py
 description = ""
 readme = "README.md"
 license = {file = "LICENSE"}
-authors = [{name = "jax_ml_stack authors", email="jax_ml_stack@google.com"}]
+authors = [{name = "jax-ai-stack authors", email="jax-dev+jax-ai-stack@google.com"}]
 classifiers = [
     "Programming Language :: Python :: 3",
     "Programming Language :: Python :: 3 :: Only",
@@ -23,8 +23,8 @@ dependencies = [
 ]
 
 [project.urls]
-homepage = "https://github.com/jax-ml/jax_ml_stack"
-repository = "https://github.com/jax-ml/jax_ml_stack"
+homepage = "https://github.com/jax-ml/jax_-ai-stack"
+repository = "https://github.com/jax-ml/jax-ai-stack"
 # Other: `documentation`, `changelog`
 
 [project.optional-dependencies]
@@ -60,5 +60,5 @@ requires = [
 ]
 
 [tool.setuptools]
-packages = ["jax_ml_stack"]
+packages = ["jax_ai_stack"]
 include-package-data = false