Cartesian-3j and Cartesian-nj Symbols
- Cartesian-3j and Cartesian-nj symbols are tensor-coupling coefficients that decompose irreducible Cartesian tensors (ICTs) for machine learning, providing a clear alternative to spherical approaches.
- They systematically extract angular-momentum blocks from tensor products using precomputed ICTD matrices, ensuring symmetry and orthonormality in the coupling process.
- Their implementation in the Python package cartnn enhances computational efficiency in low-rank ICT operations, optimizing performance in equivariant neural network architectures.
The Cartesian-3j and Cartesian-nj symbols are tensor-coupling coefficients that generalize the landscape of equivariant computations in atomistic machine learning, providing a direct analogue to the widely used Wigner-3j and Wigner-nj symbols from spherical tensor theory. These symbols enable the combination of irreducible Cartesian tensors—ICTs—into new irreducible representations, thereby furnishing an alternative to spherical-based approaches. This formalism underpins tensor product and contraction operations in Cartesian space, facilitates rigorous comparisons between Cartesian and spherical architectures, and is implemented in the Python package cartnn as an extension of e3nn (Xu et al., 18 Dec 2025).
1. Definitions and Structural Role
An irreducible Cartesian tensor (ICT) of rank is a representation of the orthogonal group of dimensionality , which is generally reducible. Decomposition yields irreducible blocks:
where each block with weight has components; distinguishes multiplicities. The tensor product of two ICTs with weights gives components of weight for .
The Cartesian-3j symbol, denoted , is the linear map extracting the irreducible component of weight from the tensor product:
or, vectorized:
The Cartesian-nj symbol generalizes this coupling to ICTs via a product of 3j symbols, constructing
with the chain , , up to -fold coupling.
2. Mathematical Formulation, Symmetry, and Orthogonality
The construction of the Cartesian-3j symbol leverages the ICTD “path” matrix to select angular-momentum blocks and re-project into the irreducible space:
The full -fold symbol is correspondingly:
using , .
Permutation symmetry is inherited from the orthogonality of ICTD matrices:
with orthonormality on input indices as
3. Algorithmic Realization and Computational Recursion
Computation proceeds as follows:
- ICTD Matrix Construction: For ranks , , build
- ;
- .
- Tensor Product Flattening: Take , flatten to .
- Projection: Project to weight using :
- Post-processing: If required, recover Cartesian representation with .
For -fold coupling, steps 2–4 are iterated recursively, treating each new projected tensor as a factor for further coupling.
4. Relation to Spherical Wigner Symbols
The role of precisely mirrors the Wigner 3-j symbol in spherical tensor coupling. Both facilitate angular-momentum block selection and abide by identical orthogonality and permutation-phase relations. The principal distinction is the ambient space dimensionality and decomposition requirements:
- Spherical basis: Each irreducible representation has dimension . Coupling is optimized by construction.
- Cartesian basis: ICTs are embedded in larger spaces, requiring ICTD decomposition to isolate angular-momentum blocks.
Both coupling diagrams yield the same admissible values, .
5. Implementation Aspects in cartnn and e3nn Ecosystem
The cartnn Python module integrates these symbols via the ICTD routines in cartnn.o3.ICTD. C-matrices and corresponding are precomputed for all on module import, typically up to . Runtime costs for coupling are
- Initial computation: , amortized over multiple uses.
- Each coupling: .
ICTD matrices are orthonormal, ensuring numerical stability up to machine precision. Memory requirements for storage scale as ; for , this remains feasible. For high or correlation order , the size of Cartesian-nj tensors can become prohibitive.
6. Explicit Example: Rank-1 ICT Coupling
Consider coupling two rank-1 ICTs and , with weights . Possible couplings are:
| Symbol | Expression | |
|---|---|---|
| $0$ | , yields | |
| $1$ | , yields | |
| $2$ |
Given :
- , other components zero.
7. Regimes and Architectures Benefiting from Cartesian Coupling
Cartesian coupling is advantageous in specific computational regimes:
- For low-rank ICTs ( or $3$), Cartesian formulas avoid spherical-harmonic evaluation and incur fewer transcendental operations.
- In edge-feature architectures where , Cartesian packaging decreases memory footprint and speeds up convolutions in architectures such as TensorNet.
- At high correlation order or large , Cartesian-nj storage costs scale combinatorially, and spherical tensors remain preferred for compactness.
- Hybrid designs may utilize Cartesian coupling for low- (local) filters, spherical for high- (global) features; this mixing offers flexibility for application-specific optimization.
The direct implementation of Cartesian-3j and Cartesian-nj in the cartnn extension of e3nn, based on precomputed orthonormal ICTD matrices, enables comprehensive support for irreducible Cartesian tensor-product operations. While low-rank cases are speed- and simplicity-optimized, high-rank or extended correlation scenarios often favor the established compactness of spherical tensor approaches (Xu et al., 18 Dec 2025).