From 57b9b9a0d33b2069b01c3ab22183be1f764ab331 Mon Sep 17 00:00:00 2001 From: erfanzar Date: Thu, 30 May 2024 13:25:32 -0400 Subject: [PATCH] using namesharding instead of partitionspec --- pyproject.toml | 2 +- src/fjformer/__init__.py | 2 +- 2 files changed, 2 insertions(+), 2 deletions(-) diff --git a/pyproject.toml b/pyproject.toml index 3715065..22fb87f 100644 --- 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 index b82b662..3614194 100644 --- 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