jax<0.4.0
jaxlib<0.4.0
jaxutils
jaxtyping>=0.0.2
jaxlinop>=0.0.3

[cuda]
jax[cuda]

[dev]
gpjax
black
isort
pylint
flake8
pytest
networkx
pytest-cov
