Small version of e3nn
- spherical harmonics
- Rs operations
- Wigner representations (up to L=11)
- 3J symbols (up to L=11)
- Gate (parity)
- radial networks
- Tensor Product (Weighted)
- message_passing (in examples)
- reduce_tensor
- tests!
- s2grid
- s2/so3 activation
- batchnorm
- dropout
- groupnorm
- image convolution (requires kernel?!)
- cuda kernels!
- tensor classes
- kernels?
- sparse tensor product
- data_helpers (contribute to
pytorch_geometric
instead) - norm activation