jax>=0.4.20
jaxlib>=0.4.20
numpy>=1.24

[dev]
pytest>=7.0
pytest-cov
