Skip to content

2023-06-23
Compare
Choose a tag to compare
@mariogeiger mariogeiger released this 23 Jun 22:05
· 128 commits to main since this release
0.19.2
6326cd9

By merging two jnp.einsum in one, the tensor product is faster than before (60% faster in the case I tested, see BENCHMARK.md).

Changelog

Changed

  • [BREAKING] Move Instruction, FunctionalTensorProduct and FunctionalFullyConnectedTensorProduct into e3nn.legacy submodule
  • Reimplement e3nn.tensor_product and e3nn.elementwise_tensor_product in a simpler way
Assets 2
Loading