e3nn API

• o3
  • Parametrization of Rotations
  • Irreps
  • Tensor Product
  • Spherical Harmonics
  • Reduction of Tensors in Irreps
  • Grid Signal on the Sphere
  • Wigner Functions
• nn
  • Gate
  • Fully Connected Neural Network
  • Batch Normalization
  • Spherical Activation
  • Norm-Based Activation
  • nn - Models
• io
  • Spherical Tensor
  • Cartesian Tensor
• math
  • direct_sum()
  • orthonormalize()
  • complete_basis()
  • soft_one_hot_linspace()
  • soft_unit_step()
• util
  • JIT - wrappers for TorchScript
  • test - helpers for unit testing