LICENSE
README.md
pyproject.toml
setup.cfg
setup.py
e3nn_jax/__init__.py
e3nn_jax/_activation.py
e3nn_jax/_batchnorm.py
e3nn_jax/_config.py
e3nn_jax/_core_tensor_product.py
e3nn_jax/_dropout.py
e3nn_jax/_einsum.py
e3nn_jax/_gate.py
e3nn_jax/_graph_util.py
e3nn_jax/_instruction.py
e3nn_jax/_irreps.py
e3nn_jax/_irreps_array.py
e3nn_jax/_linear.py
e3nn_jax/_nn.py
e3nn_jax/_poly_envelope.py
e3nn_jax/_rotation.py
e3nn_jax/_so3.py
e3nn_jax/_soft_one_hot_linspace.py
e3nn_jax/_spherical_harmonics.py
e3nn_jax/_su2.py
e3nn_jax/_tensor_products.py
e3nn_jax/perm.py
e3nn_jax.egg-info/PKG-INFO
e3nn_jax.egg-info/SOURCES.txt
e3nn_jax.egg-info/dependency_links.txt
e3nn_jax.egg-info/requires.txt
e3nn_jax.egg-info/top_level.txt
e3nn_jax/experimental/__init__.py
e3nn_jax/experimental/point_convolution.py
e3nn_jax/experimental/transformer.py
e3nn_jax/experimental/voxel_convolution.py
e3nn_jax/experimental/voxel_pooling.py
e3nn_jax/irrep/__init__.py
e3nn_jax/util/__init__.py
e3nn_jax/util/decorators.py
e3nn_jax/util/jit.py
e3nn_jax/util/sympy.py
e3nn_jax/util/test.py