diff --git a/pyproject.toml b/pyproject.toml
index 8eed9dd..ba2b2ba 100644
--- a/pyproject.toml
+++ b/pyproject.toml
@@ -4,7 +4,7 @@ build-backend = "hatchling.build"
 
 [project]
 name = "google_metrax"
-version = "0.2.1"
+version = "0.2.2"
 authors = [
   { name="Jiwon Shin", email="jshin1394@gmail.com" },
 ]
@@ -17,11 +17,11 @@ classifiers = [
 ]
 license = {file = 'LICENSE'}
 dependencies = [
-  "clu==0.0.12",
+  "clu>=0.0.12",
   "flax>=0.11.1",
-  "jax==0.6.2",
-  "numpy==2.1.3",
-  "tensorboardX==2.6.4",
+  "jax>=0.6.2",
+  "numpy>=2.1.3",
+  "tensorboardX>=2.6.4",
 ]
 
 [tool.hatch.build]