/torch-metis

Version of pytorch-metis that has bindings that directly accept pytorch tensors for constructing metis graphs. No data copies are required to convert from pytorch CSR and Metis's graph representation because METIS uses CSR internally.

Primary LanguagePythonOtherNOASSERTION

Stargazers