GitHub repository: zrqiao/torch-gauge
Primary language: Python
Description: A light-weight PyTorch extension for equivariant deep learning