This library aims to help the development of equivariant polynomials on reductive Lie groups and finite groups. It contains fundamental mathematical operations such as tensor products, symmetric products, and direct sums of reducible and irreducible representations of groups.
The library is modular, and new groups can be easily implemented.
To install via pip, follow the steps below:
git clone git@github.com:lie-nn/lie-nn.git
pip install ./lie-nn
Irreducible representations (irreps) of reductive Lie groups play an essential role as they are budling blocks of other finite-dimensional representations. We implement irreducible representations for a variety of Lie groups by providing the following:
- labelling of the irreps
- their dimensions
- an explicit basis (called generators)
- direct sums of irreps
- decompositions of tensor product of irreps into irreps (Clebsch Gordans)
- symmetric power of irreps (Symmetric generalized Clebsch Clebsch Gordans)
lie-nn currently has tabulated the irreps of the following groups:
| Groups | Potential Applications |
|---|---|
| 3D Point Clouds (Molecules, Vision) | |
| Rotations + Reflections | |
| Particles Physics | |
| Quantum Physics | |
| QCD | |
| Electromagnetism | |
| - |
Irreps of a product of all these groups are also supported. Moreover isomorphics groups have the same irreducible representations. As all the irreps of
We aim to add new tabulated irreps for more groups, including the other classical complex Lie groups
The data are organised in three categories:
GenericRep- a generic representation of a Lie groupReducedRep- a reduced representation into irrepsRep- a reducible representations (Sn,so(2n)_adjoint,so(2n+1)_adjoint)TabulatedIrrep- a tabulated irreps (SU2,SU2Real,O3,SO3,SL2C,SO13,SU3,SU4,Z2)
-
change_basis- change of basis for the representation -
change_algebra- change of basis for the algebra -
infer_change_of_basis- infer the change of basis matrix between two representations -
clebsch_gordan- Clebsch-Gordan coefficients -
reduce- creates a reduced representation (ReducedRep) -
direct_sum- direct sum of two representations -
tensor_product- tensor product of two representations -
reduced_tensor_product_basis- change of basis matrix between the tensor product and the irreps -
reduced_symmetric_tensor_product_basis- change of basis matrix between the symmetric tensor product and the irreps -
conjugate- conjugate representation -
make_explicitly_real- some representations can be made explicitly real -
group_product- representation in the direct product of the groups
import lie_nn as lie
lie.clebsch_gordan(
lie.irreps.SU3((1, 1, 0)),
lie.irreps.SU3((1, 0, 0)),
lie.irreps.SU3((2, 1, 0)),
)pycln .
black .
If you use this code, please cite our papers:
@misc{batatia2023general,
title={A General Framework for Equivariant Neural Networks on Reductive Lie Groups},
author={Ilyes Batatia and Mario Geiger and Jose Munoz and Tess Smidt and Lior Silberman and Christoph Ortner},
year={2023},
eprint={2306.00091},
archivePrefix={arXiv},
primaryClass={stat.ML}
}