Equitorch is a modularized package that can be used to flexibly constructing O(3)/SO(3) equivariant neural networks.
This package is still under development. We are actively adding more operations, documentations and tutorials.
This package is based on Pytorch(>=2.4), Pytorch-Geometric(>=2.4), and Triton(>=3.2). Please make sure you have already installed the version that fit your device. (It is currently recommended to use pip to install the Pytorch-Geometric.)
With these packages installed, you can install Equitorch by
pip install git+https://github.com/GTML-LAB/Equitorch.git