From 8618490a3dbafae17fdd5a525a25991e502132aa Mon Sep 17 00:00:00 2001
From: Joaquin Matres <4514346+joamatab@users.noreply.github.com>
Date: Tue, 9 Apr 2024 17:30:37 -0700
Subject: [PATCH] pin min version of flax

---
 pyproject.toml | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/pyproject.toml b/pyproject.toml
index c2e3061..a71724f 100644
--- a/pyproject.toml
+++ b/pyproject.toml
@@ -27,7 +27,7 @@ classifiers = [
 dependencies = [
   "black",
   "fastcore",
-  "flax",
+  "flax>=0.8.2",
   "jax",
   "jaxlib",
   "jaxtyping",