Equitorch documentation

This is Equitorch, a modularized package for flexibily constructing equivariant (and invariant) neural networks with Triton accelerated operators.

Github pages: https://github.com/GTML-LAB/Equitorch

• API Reference
  • equitorch.constants
  • equitorch.irreps
  • equitorch.nn
  • equitorch.structs
  • equitorch.transforms
  • equitorch.utils