Skip to content

2023-03-15

Compare
Choose a tag to compare
@mariogeiger mariogeiger released this 15 Mar 17:27
· 280 commits to main since this release

⚠️ e3nn.tensor_product_with_spherical_harmonics backward propagation is not yet working.

Added

  • e3nn.dot: compute the dot product between two IrrepsArray
  • per_irrep argument to e3nn.norm: compute the norm of each irrep independently if per_irrep=True
  • e3nn.tensor_product_with_spherical_harmonics from https://arxiv.org/pdf/2302.03655.pdf

Changed

  • __repr__(Irreps()) has been changed from "" to "Irreps()"

Fixed

  • spherical harmonics edge case when output_irreps=Irreps()