LICENSE
README.md
pyproject.toml
setup.cfg
setup.py
e3nn_jax/__init__.py
e3nn_jax/perm.py
e3nn_jax/util.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/_src/__init__.py
e3nn_jax/_src/activation.py
e3nn_jax/_src/activation_test.py
e3nn_jax/_src/batchnorm.py
e3nn_jax/_src/batchnorm_test.py
e3nn_jax/_src/config.py
e3nn_jax/_src/config_test.py
e3nn_jax/_src/core_tensor_product.py
e3nn_jax/_src/core_tensor_product_test.py
e3nn_jax/_src/dropout.py
e3nn_jax/_src/dropout_test.py
e3nn_jax/_src/einsum.py
e3nn_jax/_src/einsum_test.py
e3nn_jax/_src/gate.py
e3nn_jax/_src/gate_test.py
e3nn_jax/_src/grad.py
e3nn_jax/_src/grad_test.py
e3nn_jax/_src/graph_util.py
e3nn_jax/_src/graph_util_test.py
e3nn_jax/_src/instruction.py
e3nn_jax/_src/irreps.py
e3nn_jax/_src/irreps_array.py
e3nn_jax/_src/irreps_array_test.py
e3nn_jax/_src/irreps_test.py
e3nn_jax/_src/linear.py
e3nn_jax/_src/linear_test.py
e3nn_jax/_src/mlp.py
e3nn_jax/_src/perm_test.py
e3nn_jax/_src/poly_envelope.py
e3nn_jax/_src/poly_envelope_test.py
e3nn_jax/_src/reduced_tensor_product.py
e3nn_jax/_src/reduced_tensor_product_test.py
e3nn_jax/_src/rotation.py
e3nn_jax/_src/rotation_test.py
e3nn_jax/_src/so3.py
e3nn_jax/_src/soft_one_hot_linspace.py
e3nn_jax/_src/spherical_harmonics.py
e3nn_jax/_src/spherical_harmonics_test.py
e3nn_jax/_src/su2.py
e3nn_jax/_src/tensor_products.py
e3nn_jax/_src/tensor_products_test.py
e3nn_jax/_src/wigner_test.py
e3nn_jax/_src/util/__init__.py
e3nn_jax/_src/util/decorators.py
e3nn_jax/_src/util/jit.py
e3nn_jax/_src/util/jit_test.py
e3nn_jax/_src/util/math_numpy.py
e3nn_jax/_src/util/optimize_jaxpr.py
e3nn_jax/_src/util/prod.py
e3nn_jax/_src/util/sympy.py
e3nn_jax/_src/util/test.py
e3nn_jax/experimental/__init__.py
e3nn_jax/experimental/point_convolution.py
e3nn_jax/experimental/point_convolution_test.py
e3nn_jax/experimental/transformer.py
e3nn_jax/experimental/transformer_test.py
e3nn_jax/experimental/voxel_convolution.py
e3nn_jax/experimental/voxel_convolution_test.py
e3nn_jax/experimental/voxel_pooling.py
e3nn_jax/experimental/voxel_pooling_test.py
e3nn_jax/irrep/__init__.py