CHANGELOG.md
LICENSE
MANIFEST.in
README.md
pyproject.toml
e3nn_backend/__init__.py
e3nn_backend/protocol.py
e3nn_backend/registry.py
e3nn_core/__init__.py
e3nn_core/cg.py
e3nn_core/instructions.py
e3nn_core/irreps.py
e3nn_core/normalization.py
e3nn_core/rotations.py
e3nn_core/typing.py
e3nn_core/wigner.py
e3nn_mlx/__init__.py
e3nn_mlx/_high_level.py
e3nn_mlx/_metal_scatter.py
e3nn_mlx/_metal_sh.py
e3nn_mlx/_metal_tp.py
e3nn_mlx/_version.py
e3nn_mlx/backend.py
e3nn_mlx/compat.py
e3nn_mlx/graph.py
e3nn_mlx/irreps_array.py
e3nn_mlx/nn_activation.py
e3nn_mlx/nn_batchnorm.py
e3nn_mlx/nn_dropout.py
e3nn_mlx/nn_extract.py
e3nn_mlx/nn_fc.py
e3nn_mlx/nn_gate.py
e3nn_mlx/nn_identity.py
e3nn_mlx/nn_linear.py
e3nn_mlx/nn_norm.py
e3nn_mlx/nn_normact.py
e3nn_mlx/nn_s2act.py
e3nn_mlx/nn_so3act.py
e3nn_mlx/ops_basic.py
e3nn_mlx/ops_reduce.py
e3nn_mlx/ops_reduce_tensor.py
e3nn_mlx/ops_rotations.py
e3nn_mlx/ops_s2.py
e3nn_mlx/ops_sh.py
e3nn_mlx/ops_so3.py
e3nn_mlx/ops_tp.py
e3nn_mlx/profiling.py
e3nn_mlx/radial.py
e3nn_mlx.egg-info/PKG-INFO
e3nn_mlx.egg-info/SOURCES.txt
e3nn_mlx.egg-info/dependency_links.txt
e3nn_mlx.egg-info/requires.txt
e3nn_mlx.egg-info/top_level.txt
e3nn_mlx/math/__init__.py
e3nn_mlx/models/__init__.py
e3nn_mlx/models/gate_points_2102.py
e3nn_mlx/models/v2106/__init__.py
e3nn_mlx/models/v2106/gate_points_message_passing.py
e3nn_mlx/models/v2106/gate_points_networks.py
e3nn_mlx/models/v2106/points_convolution.py
e3nn_mlx/nn/__init__.py
e3nn_mlx/nn/models/__init__.py
e3nn_mlx/nn/models/gate_points_2102.py
e3nn_mlx/nn/models/v2106/__init__.py
e3nn_mlx/nn/models/v2106/gate_points_message_passing.py
e3nn_mlx/nn/models/v2106/gate_points_networks.py
e3nn_mlx/nn/models/v2106/points_convolution.py
e3nn_mlx/o3/__init__.py
tests/test_backend_registry.py
tests/test_cg_and_wigner_core.py
tests/test_evals.py
tests/test_gate_points_2102_convolution.py
tests/test_gate_points_2102_network.py
tests/test_graph_radial.py
tests/test_high_level_api.py
tests/test_instructions.py
tests/test_irreps.py
tests/test_irreps_array.py
tests/test_metal_fallbacks.py
tests/test_metal_kernels.py
tests/test_mlx_module_integration.py
tests/test_mlx_module_state.py
tests/test_modules.py
tests/test_nn_import_policy.py
tests/test_p0_equivariance.py
tests/test_package_metadata.py
tests/test_profiling_and_extensions.py
tests/test_reference_harness.py
tests/test_rotations.py
tests/test_spherical_harmonics.py
tests/test_tensor_product.py
tests/test_tensor_product_correctness.py
tests/test_tensor_product_wrappers.py
tests/test_upstream_nn_basic.py
tests/test_upstream_nn_gate.py
tests/test_upstream_nn_norm_fc.py
tests/test_upstream_nn_s2act.py
tests/test_upstream_nn_so3act.py
tests/test_upstream_o3_irreps_wigner.py
tests/test_upstream_o3_linear_norm.py
tests/test_upstream_o3_reduce_tensor.py
tests/test_upstream_o3_rotations_sh.py
tests/test_upstream_o3_s2.py
tests/test_upstream_o3_tensor_product.py
tests/test_v2106_convolution.py
tests/test_v2106_message_passing.py
tests/test_v2106_networks.py