Releases: gerkone/segnn-jax
Releases · gerkone/segnn-jax
segnn-jax 0.7
- adapted to the newest breaking changes from
e3nn 0.19.0
- added
O3TensorProductSCN
usinge3nn.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
- Updated to jax 0.4.8 and e3nn-jax 0.17.4
- Simplified tensor product interface
- Improved QM9 validation results
segnn-jax 0.5
- 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
SEGNN
as anhk.Module
class to allow easier extensions- updated readme
segnn-jax 0.2
updated readme Former-commit-id: 5eca88f2ffaff4e1aae81d88ba4bab0bebc3d377