jax>=0.3.7
jaxlib>=0.1.69

[testing]
flake8
pylint
pytest
pytype
