zrqiao / torch-gauge

A light-weight PyTorch extension for equivariant deep learning
GNU General Public License v3.0
15 stars 8 forks source link