staradutt's Stars
NVIDIA/cuEquivariance
cuEquivariance is a math library that is a collective of low-level primitives and tensor ops to accelerate widely-used models, like DiffDock, MACE, Allegro and NEQUIP, based on equivariant neural networks.
FilippoMB/Spectral-Clustering-with-Graph-Neural-Networks-for-Graph-Pooling
Reproduces the results of MinCutPool as presented in the 2020 ICML paper "Spectral Clustering with Graph Neural Networks for Graph Pooling".