numpy~=1.18
scipy~=1.5
tqdm~=4.56
plum-dispatch~=1.0
numba<0.54,>=0.52
networkx~=2.4
jax<0.2.14,>=0.2.9
jaxlib>=0.1.57
flax<0.4,>=0.3.0
orjson~=3.4
optax<=0.0.7,>=0.0.2

[all]
mpi4py>=3.0.1
mpi4jax>=0.2.11
pytest>=6
pytest-xdist>=2
coverage>=5
pytest-cov>=2.10.1
python-igraph
pre-commit
black==20.8b1
tensorboardx>=2.0.0

[dev]
pytest>=6
pytest-xdist>=2
coverage>=5
pytest-cov>=2.10.1
python-igraph
pre-commit
black==20.8b1

[mpi]
mpi4py>=3.0.1
mpi4jax>=0.2.11

[tensorboard]
tensorboardx>=2.0.0
