From 3aa4bd09531422b2c48f345e79124ce0037b9344 Mon Sep 17 00:00:00 2001
From: bskrlj <bskrlj@outbrain.com>
Date: Mon, 13 May 2024 13:33:30 +0200
Subject: [PATCH] Reqs

---
 requirements.txt | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/requirements.txt b/requirements.txt
index 4b674f3..bdbebc9 100644
--- a/requirements.txt
+++ b/requirements.txt
@@ -1,5 +1,5 @@
 flake8>=6.1.0
-flux==1.3.5
+flax==1.3.5
 jax==0.4.28
 jaxlib==0.4.28
 matplotlib>=3.7.2