safetensors>=0.4.0
numpy>=1.24
jax>=0.4.0

[all]
torch>=2.0
flax>=0.7
ml_dtypes

[dev]
pytest
black
ruff

[flax]
flax>=0.7

[torch]
torch>=2.0
