numpy>=1.21.0
pandas>=1.3.0
scipy>=1.7.0
scikit-learn>=1.0.0
matplotlib>=3.3.0
seaborn>=0.11.0
jax>=0.3.0

[dev]
pytest>=6.0
pytest-cov
black
flake8
mypy
jupyter

[gpu]
jax[cuda]>=0.3.0
