There is a newer version of the record available.

Published December 16, 2021 | Version 0.4.4
Software Open

e3nn/e3nn: 2021-12-15

  • 1. @atomicarchitects
  • 2. @mir-group @ Harvard University
  • 3. University of Amsterdam
  • 4. @NVIDIA
  • 5. Harvard
  • 6. Technical University of Denmark (DTU)
  • 7. @seoklab @Galux-Inc
  • 8. Redwood Center for Theoretical Neuroscience
  • 9. University of California, Berkeley

Description

[0.4.4] - 2021-12-15 Fixed

  • Remove CartesianTensor._rtp. Instead recompute the ReducedTensorProduct everytime. The user can save the ReducedTensorProduct to avoid creating it each time.
  • *equivariance_error no longer keeps around unneeded autograd graphs
  • CartesianTensor builds ReducedTensorProduct with correct device/dtype when called without one
Added
  • Created module for reflected imports allowing for nice syntax for creating irreps, e.g. from e3nn.o3.irreps import l3o # same as Irreps("o3")
  • Add uvu<v mode for TensorProduct. Compute only the upper triangular part of the uv terms.
  • (beta) TensorSquare. computes x \otimes x and decompose it.
  • *equivariance_error now tell you which arguments had which error
Changed
  • Give up the support of python 3.6, set python_requires='>=3.7' in setup
  • Optimize a little bit ReducedTensorProduct: solve linear system only once per irrep instead of 2L+1 times.
  • Do not scale line width by path_weight in TensorProduct.visualize
  • *equivariance_error now transforms its inputs in float64 by default, regardless of the dtype used for the calculation itself

Files

e3nn/e3nn-0.4.4.zip

Files (755.7 kB)

Name Size Download all
md5:c0d8a3885cd0c8970b6712c8fbc801d0
755.7 kB Preview Download

Additional details

Related works