datasets>=4.6.1
flax>=0.12.5
grain>=0.2.16
jax>=0.9.1
kagglehub>=1.0.0
optax>=0.2.6
orbax>=0.1.9
safetensors>=0.7.0
huggingface-hub>=1.6.0

[dev]
pytest>=9.0.2
twine>=6.0.0
build>=1.2.0
