From 4150aeb13952755ce9be19f021cb96e71524b1dc Mon Sep 17 00:00:00 2001
From: saifxyzyz <saifmujeeb2603@gmail.com>
Date: Sat, 25 Oct 2025 20:07:45 +0530
Subject: [PATCH] modify requirements.txt

---
 requirements.txt | 6 ++++--
 1 file changed, 4 insertions(+), 2 deletions(-)

diff --git a/requirements.txt b/requirements.txt
index 3c07a14..eccd10b 100644
--- a/requirements.txt
+++ b/requirements.txt
@@ -1,5 +1,7 @@
 jax
-orbax
+orbax-export
+orbax-checkpoint
+tensorflow
 seqio
 einops
 absl-py
@@ -9,4 +11,4 @@ tensorboard
 # Following packages are only required for running simply/utils/math_eval.py
 sympy
 pylatexenc
-pytest
\ No newline at end of file
+pytest