Fix requirements.txt for Python 3.10 compatibility

- Use flexible version ranges for better compatibility
- Add jaxlib explicitly (required by jax)
- Update version constraints to avoid build conflicts

requirements.txt

@@ -1,20 +1,21 @@
 torch>=1.12.0
 transformers==4.17.0
 tokenizers==0.11.6
-numpy==1.22.0
-pandas==1.4.0
-scipy==1.8.0
-scikit-learn==1.0.0
+numpy>=1.21.0,<1.24.0
+pandas>=1.3.0,<2.0.0
+scipy>=1.7.0,<1.11.0
+scikit-learn>=1.0.0,<1.2.0
 tqdm
-biopython==1.78
+biopython>=1.78,<2.0.0
 matplotlib
 seaborn
 gradio==3.50.0
 datasets==1.18.3
 wandb==0.12.11
-jax==0.3.0
-flax==0.4.0
+jax>=0.3.0,<0.4.0
+jaxlib>=0.3.0,<0.4.0
+flax>=0.4.0,<0.5.0
 optax
 sentencepiece==0.1.96
-aiohttp==3.8.0
-sentry-sdk==1.3.1
+aiohttp>=3.8.0,<3.9.0
+sentry-sdk>=1.3.1,<2.0.0