numpy>=1.18
numba
mpi4py
torch
networkx>=2.6.2
pyyaml
Cython

[:python_version < "3.8.0"]
typing_extensions
