numpy~=1.18
scipy~=1.5
tqdm~=4.56
plum-dispatch~=1.5.1
numba<0.55,>=0.52
python-igraph~=0.9
jax<0.2.20,>0.2.16
jaxlib>=0.1.69
flax<0.4,>=0.3.0
orjson~=3.4
optax<0.0.10,>=0.0.2

[all]
mpi4py<4,>=3.0.1
mpi4jax~=0.3.1
pytest>=6
pytest-xdist>=2
coverage>=5
pytest-cov>=2.10.1
networkx~=2.4
flaky>=3.7
pre-commit
black==21.6b0
flakehell>=0.9
tensorboardx>=2.0.0

[dev]
pytest>=6
pytest-xdist>=2
coverage>=5
pytest-cov>=2.10.1
networkx~=2.4
flaky>=3.7
pre-commit
black==21.6b0
flakehell>=0.9

[mpi]
mpi4py<4,>=3.0.1
mpi4jax~=0.3.1

[tensorboard]
tensorboardx>=2.0.0
