.gitignore
.pre-commit-config.yaml
CHANGELOG.md
LICENSE
README.md
pyproject.toml
.github/workflows/docs.yaml
.github/workflows/release.yaml
.github/workflows/tests.yaml
docs/Makefile
docs/source/api.rst
docs/source/conf.py
docs/source/index.rst
docs/source/lloca-vs-lgatr.rst
docs/source/numerics.rst
docs/source/quickstart.rst
docs/source/_static/lloca.png
docs/source/_templates/autosummary/class.rst
docs/source/more-backbones/graphnet.rst
docs/source/more-backbones/index.rst
docs/source/more-backbones/particlenet.rst
docs/source/more-backbones/particletransformer.rst
docs/source/more-backbones/transformer.rst
examples/demo_transformer.ipynb
img/lloca.png
lloca/__init__.py
lloca.egg-info/PKG-INFO
lloca.egg-info/SOURCES.txt
lloca.egg-info/dependency_links.txt
lloca.egg-info/entry_points.txt
lloca.egg-info/requires.txt
lloca.egg-info/top_level.txt
lloca/backbone/__init__.py
lloca/backbone/attention.py
lloca/backbone/graphnet.py
lloca/backbone/lloca_message_passing.py
lloca/backbone/mlp.py
lloca/backbone/particlenet.py
lloca/backbone/particletransformer.py
lloca/backbone/transformer.py
lloca/backbone/attention_backends/__init__.py
lloca/backbone/attention_backends/flash.py
lloca/backbone/attention_backends/flex.py
lloca/backbone/attention_backends/native.py
lloca/backbone/attention_backends/xformers.py
lloca/equivectors/__init__.py
lloca/equivectors/base.py
lloca/equivectors/lgatr.py
lloca/equivectors/mlp.py
lloca/equivectors/pelican.py
lloca/framesnet/__init__.py
lloca/framesnet/equi_frames.py
lloca/framesnet/frames.py
lloca/framesnet/nonequi_frames.py
lloca/reps/__init__.py
lloca/reps/tensorreps.py
lloca/reps/tensorreps_transform.py
lloca/utils/__init__.py
lloca/utils/lorentz.py
lloca/utils/orthogonalize_3d.py
lloca/utils/orthogonalize_4d.py
lloca/utils/polar_decomposition.py
lloca/utils/rand_transforms.py
lloca/utils/utils.py
tests/__init__.py
tests/constants.py
tests/helpers.py
tests/hep.py
tests/lloca/__init__.py
tests/lloca/backbone/__init__.py
tests/lloca/backbone/test_attention.py
tests/lloca/backbone/test_graphnet.py
tests/lloca/backbone/test_particlenet.py
tests/lloca/backbone/test_particletransformer.py
tests/lloca/backbone/test_transformer.py
tests/lloca/equivectors/__init__.py
tests/lloca/equivectors/test_lgatr.py
tests/lloca/equivectors/test_mlp.py
tests/lloca/equivectors/test_pelican.py
tests/lloca/framesnet/__init__.py
tests/lloca/framesnet/test_equi_frames.py
tests/lloca/framesnet/test_frames.py
tests/lloca/framesnet/test_nonequi_frames.py
tests/lloca/reps/__init__.py
tests/lloca/reps/test_tensorreps_transform.py
tests/lloca/utils/__init__.py
tests/lloca/utils/test_lorentz.py
tests/lloca/utils/test_orthogonalize_3d.py
tests/lloca/utils/test_orthogonalize_4d.py
tests/lloca/utils/test_polar_decomposition.py
tests/lloca/utils/test_rand_transforms.py
tests/lloca/utils/test_utils.py