jax>=0.3.14
jaxlib>=0.3.14

[dev]
pytest
pytest-cov
pytest-pep8
