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
tokenizers>=0.22.2
huggingface-hub>=1.6.0
wandb>=0.25.1
tensorboard>=2.20.0
torch>=2.10.0

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