Skip to content

segnn-jax 0.7

Latest
Compare
Choose a tag to compare
@gerkone gerkone released this 16 Jul 21:11
· 17 commits to main since this release
a9f8af0
  • adapted to the newest breaking changes from e3nn 0.19.0
  • added O3TensorProductSCN using e3nn.experimental.linear_shtp.shtp and a linear layer (not fully integrated)
  • removed pooling_irreps from segnn (graph task case)
  • changed the O3 layer structure
  • fixed the validation script (mistake in the loss wrapper function)