numpy<1.24.0,>=1.18.0
numba
mpi4py
torch
networkx>=2.6.2
pyyaml
Cython

[:python_version < "3.8.0"]
typing_extensions
