.gitignore
.pre-commit-config.yaml
.readthedocs.yml
CITATION.bib
INSTALL.md
LICENSE
README.md
pyproject.toml
.github/CHANGELOG.md
.github/CODE_OF_CONDUCT.md
.github/CONTRIBUTING.md
.github/PULL_REQUEST_TEMPLATE.md
.github/ISSUE_TEMPLATE/bug-report.md
.github/ISSUE_TEMPLATE/config.yml
.github/ISSUE_TEMPLATE/feature_request.md
.github/workflows/release.yml
.github/workflows/tests.yml
docs/Makefile
docs/conf.py
docs/index.rst
docs/make.bat
docs/requirements.txt
docs/api/e3nn.rst
docs/api/io/cartesian_tensor.rst
docs/api/io/io.rst
docs/api/io/spherical_tensor.rst
docs/api/math/math.rst
docs/api/nn/nn.rst
docs/api/nn/nn_bn.rst
docs/api/nn/nn_fc.rst
docs/api/nn/nn_gate.rst
docs/api/nn/nn_normact.rst
docs/api/nn/nn_s2act.rst
docs/api/nn/models/gate_points_2101.rst
docs/api/nn/models/graph.svg
docs/api/nn/models/models.rst
docs/api/nn/models/v2103.rst
docs/api/o3/o3.rst
docs/api/o3/o3_irreps.rst
docs/api/o3/o3_reduce.rst
docs/api/o3/o3_rotation.rst
docs/api/o3/o3_s2grid.rst
docs/api/o3/o3_sh.rst
docs/api/o3/o3_tp.rst
docs/api/o3/o3_wigner.rst
docs/api/util/jit.rst
docs/api/util/test.rst
docs/api/util/util.rst
docs/examples/examples.rst
docs/examples/tetris_gate.rst
docs/examples/tetris_polynomial.rst
docs/guide/change_of_basis.rst
docs/guide/convolution.rst
docs/guide/equivar_testing.rst
docs/guide/guide.rst
docs/guide/installation.md
docs/guide/irreps.rst
docs/guide/jit.rst
docs/guide/normalization.rst
docs/guide/periodic_boundary_conditions.rst
docs/guide/transformer.png
docs/guide/transformer.rst
e3nn/__init__.py
e3nn.egg-info/PKG-INFO
e3nn.egg-info/SOURCES.txt
e3nn.egg-info/dependency_links.txt
e3nn.egg-info/requires.txt
e3nn.egg-info/top_level.txt
e3nn/io/__init__.py
e3nn/io/_cartesian_tensor.py
e3nn/io/_spherical_tensor.py
e3nn/math/__init__.py
e3nn/math/_linalg.py
e3nn/math/_normalize_activation.py
e3nn/math/_reduce.py
e3nn/math/_soft_one_hot_linspace.py
e3nn/math/_soft_unit_step.py
e3nn/math/perm.py
e3nn/nn/__init__.py
e3nn/nn/_activation.py
e3nn/nn/_batchnorm.py
e3nn/nn/_dropout.py
e3nn/nn/_extract.py
e3nn/nn/_fc.py
e3nn/nn/_gate.py
e3nn/nn/_identity.py
e3nn/nn/_normact.py
e3nn/nn/_s2act.py
e3nn/nn/_so3act.py
e3nn/nn/models/__init__.py
e3nn/nn/models/gate_points_2101.py
e3nn/nn/models/gate_points_2102.py
e3nn/nn/models/v2103/__init__.py
e3nn/nn/models/v2103/conv_points_in_out.py
e3nn/nn/models/v2103/gate_points_message_passing.py
e3nn/nn/models/v2103/gate_points_networks.py
e3nn/nn/models/v2103/points_convolution.py
e3nn/nn/models/v2103/voxel_convolution.py
e3nn/nn/models/v2104/__init__.py
e3nn/nn/models/v2104/voxel_convolution.py
e3nn/nn/models/v2106/__init__.py
e3nn/nn/models/v2106/gate_points_message_passing.py
e3nn/nn/models/v2106/gate_points_networks.py
e3nn/nn/models/v2106/points_convolution.py
e3nn/nn/models/v2203/sparse_voxel_convolution.py
e3nn/o3/__init__.py
e3nn/o3/_angular_spherical_harmonics.py
e3nn/o3/_irreps.py
e3nn/o3/_linear.py
e3nn/o3/_norm.py
e3nn/o3/_reduce.py
e3nn/o3/_rotation.py
e3nn/o3/_s2grid.py
e3nn/o3/_so3grid.py
e3nn/o3/_spherical_harmonics.py
e3nn/o3/_wigner.py
e3nn/o3/_tensor_product/__init__.py
e3nn/o3/_tensor_product/_codegen.py
e3nn/o3/_tensor_product/_instruction.py
e3nn/o3/_tensor_product/_sub.py
e3nn/o3/_tensor_product/_tensor_product.py
e3nn/o3/experimental/__init__.py
e3nn/o3/experimental/_elementwise_tp.py
e3nn/o3/experimental/_full_tp.py
e3nn/o3/irrep/__init__.py
e3nn/util/__init__.py
e3nn/util/_argtools.py
e3nn/util/_context.py
e3nn/util/datatypes.py
e3nn/util/default_type.py
e3nn/util/jit.py
e3nn/util/test.py
e3nn/util/codegen/__init__.py
e3nn/util/codegen/_mixin.py
examples/README.md
examples/atom_types.py
examples/tensor_product_benchmark.py
examples/tensor_product_profile.py
examples/tetris.py
examples/tetris_gate.py
examples/tetris_polynomial.py
examples/s2cnn/mnist/README.md
examples/s2cnn/mnist/gendata.py
examples/s2cnn/mnist/train.py
tests/conftest.py
tests/defaults_test.py
tests/math/normalize_activation_test.py
tests/math/perm_test.py
tests/math/soft_one_hot_test.py
tests/math/soft_unit_step_test.py
tests/nn/activation_test.py
tests/nn/batchnorm_test.py
tests/nn/dropout_test.py
tests/nn/extract_test.py
tests/nn/fc_test.py
tests/nn/gate_test.py
tests/nn/normact_test.py
tests/nn/s2act_test.py
tests/nn/so3act_test.py
tests/nn/models/gate_points_2101_test.py
tests/nn/models/gate_points_2102_test.py
tests/nn/models/v2203/sparse_voxel_convolution_test.py
tests/o3/angular_spherical_harmonics_test.py
tests/o3/cartesian_spherical_harmonics_test.py
tests/o3/irreps_test.py
tests/o3/linear_test.py
tests/o3/norm_test.py
tests/o3/reduce_tensor_test.py
tests/o3/rotation_test.py
tests/o3/s2_test.py
tests/o3/tensor_product_sub_test.py
tests/o3/tensor_product_test.py
tests/o3/wigner_test.py
tests/o3/experimental/benchmark_pt2.py
tests/o3/experimental/test_elementwise_tp.py
tests/o3/experimental/test_fulltp.py
tests/util/test_jit.py
tests/util/test_test.py