.gitignore
LICENSE
README.md
pyproject.toml
requirements.txt
.github/workflows/release.yaml
.github/workflows/tests.yaml
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/default_attention.py
lloca/backbone/attention_backends/flex_attention.py
lloca/backbone/attention_backends/xformers_attention.py
lloca/equivectors/__init__.py
lloca/equivectors/base.py
lloca/equivectors/equimlp.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_equimlp.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