jax>=0.5.0
jaxlib>=0.5.0

[dev]
pytest
pytest-cov
pytest-pep8
