numpy>=1.20.0
jax>=0.4.0
jaxlib>=0.4.0
scipy>=1.7.0

[dev]
pytest>=7.0.0
pytest-cov>=4.0.0
matplotlib>=3.5.0
tqdm>=4.60.0
black>=22.0.0
flake8>=5.0.0
mypy>=1.0.0

[gpu]
jax[cuda12]>=0.4.0
