There is a newer version of the record available.

Published November 8, 2021 | Version 0.4.2
Software Open

e3nn/e3nn: 2021-11-08

Creators

  • 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.2] - 2021-11-08

This release coupled with the release of opt-einsum-fx=0.1.4 aim to fix slowness in the instantiation of TensorProduct. The two main change that improved the instantiation time are

  • Turning off the compilation of TensorProduct.right by default
  • Replacing actual computation of torch.einsum and torch.tensordot by prediction of their output shape in the tracer used by opt-einsum-fx to collect tensor shapes
Added
  • irrep_normalization and path_normalization for TensorProduct
  • compile_right flag to TensorProduct
  • Add new global flag jit_script_fx to optionally turn off torch.jit.script of fx code

Files

e3nn/e3nn-0.4.2.zip

Files (751.6 kB)

Name Size Download all
md5:a39964222e11c199a084c2d6c0a5108f
751.6 kB Preview Download

Additional details

Related works

Is supplement to
https://github.com/e3nn/e3nn/tree/0.4.2 (URL)