Skip to content

Releases: gerkone/segnn-jax

segnn-jax 0.7

16 Jul 21:11
a9f8af0
Compare
Choose a tag to compare
  • 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)

segnn-jax 0.6

08 May 09:44
Compare
Choose a tag to compare
  • Updated to jax 0.4.8 and e3nn-jax 0.17.4
  • Simplified tensor product interface
  • Improved QM9 validation results

segnn-jax 0.5

31 Mar 17:38
7fb05da
Compare
Choose a tag to compare
  • Updated tensor product, ~5-10% faster in training (e3nn-jax 0.14.0)
  • Fixed a bug on normalization parametrization
  • Module refactor
  • Updated results

segnn-jax 0.3

20 Jan 21:00
Compare
Choose a tag to compare
  • SEGNN as an hk.Module class to allow easier extensions
  • updated readme

segnn-jax 0.2

15 Dec 16:23
Compare
Choose a tag to compare
updated readme

Former-commit-id: 5eca88f2ffaff4e1aae81d88ba4bab0bebc3d377