torch>=1.12.0
transformers>=4.30.0
xgboost>=1.7.0
numpy>=1.20
pandas>=1.3
scikit-learn>=1.0
jax-unirep>=2.0.0
wandb>=0.14
tomli>=0.10.2

[dev]
pytest>=6.0
black>=22.0
flake8>=3.9

[ui]
streamlit>=1.18
matplotlib>=3.5
