using namesharding instead of partitionspec

erfanzar · May 30, 2024 · 57b9b9a · 57b9b9a
1 parent 1acbd05
commit 57b9b9a
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/pyproject.toml b/pyproject.toml
@@ -5,7 +5,7 @@ authors = [
 ]
 requires-python = ">=3.8"
 readme = "README.md"
-version = "0.0.63"
+version = "0.0.64"
 
 dependencies = [
     "jax>=0.4.20",

diff --git a/src/fjformer/__init__.py b/src/fjformer/__init__.py
@@ -68,7 +68,7 @@
 from . import optimizers as optimizers
 from . import linen as linen
 
-__version__ = "0.0.63"
+__version__ = "0.0.64"
 
 __all__ = (
     # Loss and extra function