apex
pytorch >= 1.7.1
torch_geometric >= 1.7.0
torch_sparse >= 0.6.9
mip >= 1.13.0