Skip to content

Releases: e3nn/e3nn-jax

2022-03-26

26 Mar 17:58

Choose a tag to compare

Added

  • IrrepsData can be given as argument of spherical_harmonics
  • added broadcasting of IrrepsData, elementwise_tensor_product, FullyConnectedTensorProduct, full_tensor_product

Changed

  • BatchNorm supports None
  • BatchNorm supports change default value of eps from 1e-5 to 1e-4
  • gate change default odd activation to (1 - exp(x^2)) * x

Full Changelog: 0.4.2...0.4.3

2022-03-23

23 Mar 18:20

Choose a tag to compare

Added

  • gate list of activations argument is now optional
  • experimental.transformer.Transformer simplified interface using IrrepsData and swap two arguments order

2022-03-21

21 Mar 20:33

Choose a tag to compare

Added

  • IrrepsData.repeat_irreps_by_last_axis
  • IrrepsData.repeat_mul_by_last_axis
  • IrrepsData.factor_mul_to_last_axis
  • add axis argument to IrrepsData.cat
  • IrrepsData.remove_nones
  • IrrepsData.ones

Changed

  • experimental.point_convolution.Convolution simplified interface using IrrepsData

2022-03-19

20 Mar 00:14

Choose a tag to compare

First release of e3nn-jax