conda create --name gaunt-ef python=3.10
conda activate gaunt-ef
pip install torch==1.13.1+cu117 -f https://download.pytorch.org/whl/torch_stable.html
pip install e3nn
pip install ipykernel
- Efficiency comparison experiment for Equivariant Feature Interaction:
test_equi_feat.ipynb
- Efficiency comparison experiment for Equivariant Convolution:
test_equi_conv.ipynb
- Efficiency comparison experiment for Equivariant Many-Body Interaction:
test_equi_many-body.ipynb