celluloid~=0.2.0
chex~=0.1.88
jaxtyping~=0.2.36
jax>=0.4.24
jaxlib>=0.4.24
numpy~=1.26.4
flax>=0.8.1
joblib~=1.3.2
transformer-payne~=0.5

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

[huggingface]
huggingface-hub~=0.21.3
