From 5f2a6fe72c51d2d203c2b6ec157e040313730bfe Mon Sep 17 00:00:00 2001
From: Fabian Pedregosa <pedregosa@google.com>
Date: Thu, 7 Mar 2024 10:53:26 -0800
Subject: [PATCH] 0.2.1 release

PiperOrigin-RevId: 613637111
---
 optax/__init__.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/optax/__init__.py b/optax/__init__.py
index a94627461..1622b0f36 100644
--- a/optax/__init__.py
+++ b/optax/__init__.py
@@ -213,7 +213,7 @@
 )
 dpsgd = contrib.dpsgd
 
-__version__ = "0.2.1.dev"
+__version__ = "0.2.1"
 
 __all__ = (
     "adabelief",