jax>=0.4.1
jaxlib>=0.4.1
jaxutils>=0.0.8
jaxtyping>=0.0.2
jaxlinop>=0.0.3
deprecation

[cuda]
jax[cuda]

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