numpy>=1.18
scipy>=1.5.2
tqdm>=4.56.2
plum-dispatch~=1.0.0
numba>=0.52.0
networkx>=2.4
jax>=0.2.9
jaxlib>=0.1.57
flax>=0.3.0
orjson>=3.4
optax>=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
