Skip to content

segnn-jax 0.5

Compare
Choose a tag to compare
@gerkone gerkone released this 31 Mar 17:38
· 37 commits to main since this release
7fb05da
  • Updated tensor product, ~5-10% faster in training (e3nn-jax 0.14.0)
  • Fixed a bug on normalization parametrization
  • Module refactor
  • Updated results