Skip to content

2022-12-07
Compare
Choose a tag to compare
@mariogeiger mariogeiger released this 07 Dec 23:49
· 429 commits to main since this release
0.13.0
187e02c 
import e3nn_jax as e3nn

irreps = e3nn.Irreps("0e + 1o")

print(3 * irreps)  # prints 3x0e+3x1o

Changed

  • [BREAKING] 3 * e3nn.Irreps("0e + 1o") now returns 3x0e + 3x1o instead of 1x0e + 1x1o + 1x0e + 1x1o + 1x0e + 1x1o
  • [BREAKING] in Linear, renamed num_weights to num_indexed_weights because it was confusing.

Added

  • e3nn.Irreps("3x0e + 6x1o") // 3 returns 1x0e + 2x1o

Fixed

  • s2grid is now jitable
Assets 2
Loading