flax>=0.12.0
jax>=0.8.0
jaxlib>=0.8.0
optax>=0.2.6

[dev]
pytest>=9.0.1
absl-py>=2.0.0
numpy>=1.26.0

[pretrained]
safetensors>=0.4.0
huggingface_hub>=0.20.0
transformers>=4.40.0
