search

e3nn / e3nn-jax

jax library for E3 Equivariant Neural Networks
Apache License 2.0
182 stars 18 forks source link

Cache cg coefficients 2 #73

Closed mitkotak closed 4 months ago

mitkotak commented 4 months ago

Resolves comments in https://github.com/e3nn/e3nn-jax/pull/71

mitkotak commented 4 months ago

@mariogeiger Can I get a review ?