jax>=0.4.0
jaxlib>=0.4.0
safetensors>=0.4.0

[dev]
pytest>=7.0
pytest-cov
black
ruff

[examples]
streamlit
numpy
matplotlib
