From fc2e5993abf4aff2835ffe1b0a509a1bc7df60e3 Mon Sep 17 00:00:00 2001
From: Cristian Garcia <cgarciae@google.com>
Date: Tue, 27 Aug 2024 10:36:45 -0700
Subject: [PATCH] update Flax to version 0.9.0

PiperOrigin-RevId: 668047408
---
 README.md       | 4 ++--
 flax/version.py | 2 +-
 2 files changed, 3 insertions(+), 3 deletions(-)

diff --git a/README.md b/README.md
index 63eda1cee..92073c711 100644
--- a/README.md
+++ b/README.md
@@ -199,8 +199,8 @@ To cite this repository:
   author = {Jonathan Heek and Anselm Levskaya and Avital Oliver and Marvin Ritter and Bertrand Rondepierre and Andreas Steiner and Marc van {Z}ee},
   title = {{F}lax: A neural network library and ecosystem for {JAX}},
   url = {http://github.com/google/flax},
-  version = {0.8.6},
-  year = {2023},
+  version = {0.9.0},
+  year = {2024},
 }
 ```
 
diff --git a/flax/version.py b/flax/version.py
index 42d01b808..dd9609eb1 100644
--- a/flax/version.py
+++ b/flax/version.py
@@ -13,4 +13,4 @@
 # limitations under the License.
 
 """Current Flax version at head on Github."""
-__version__ = '0.8.6'
+__version__ = '0.9.0'