jax~=0.4.24
jaxlib~=0.4.24
numpy~=1.26.4
flax~=0.8.1

[dev]
pytest~=8.0.0
pytest-datadir~=1.5.0
coverage~=7.4.3

[huggingface]
huggingace-hub~=0.20.3
joblib~=1.3.2
