.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/attention_backends.rst
docs/source/conf.py
docs/source/geometric_algebra.rst
docs/source/index.rst
docs/source/quickstart.rst
docs/source/quickstart_slim.rst
docs/source/symmetry_breaking.rst
docs/source/_static/gatr.png
docs/source/_templates/autosummary/class.rst
examples/demo_conditional_lgatr.ipynb
examples/demo_lgatr.ipynb
examples/demo_lgatr_slim.ipynb
img/gatr.png
img/logo.png
lgatr/__init__.py
lgatr.egg-info/PKG-INFO
lgatr.egg-info/SOURCES.txt
lgatr.egg-info/dependency_links.txt
lgatr.egg-info/entry_points.txt
lgatr.egg-info/requires.txt
lgatr.egg-info/top_level.txt
lgatr/interface/__init__.py
lgatr/interface/axialvector.py
lgatr/interface/pseudoscalar.py
lgatr/interface/scalar.py
lgatr/interface/spurions.py
lgatr/interface/vector.py
lgatr/layers/__init__.py
lgatr/layers/conditional_lgatr_block.py
lgatr/layers/dropout.py
lgatr/layers/layer_norm.py
lgatr/layers/lgatr_block.py
lgatr/layers/linear.py
lgatr/layers/attention/__init__.py
lgatr/layers/attention/attention.py
lgatr/layers/attention/config.py
lgatr/layers/attention/cross_attention.py
lgatr/layers/attention/qkv.py
lgatr/layers/attention/self_attention.py
lgatr/layers/mlp/__init__.py
lgatr/layers/mlp/config.py
lgatr/layers/mlp/geometric_bilinears.py
lgatr/layers/mlp/mlp.py
lgatr/layers/mlp/nonlinearities.py
lgatr/nets/__init__.py
lgatr/nets/conditional_lgatr.py
lgatr/nets/conditional_lgatr_slim.py
lgatr/nets/lgatr.py
lgatr/nets/lgatr_slim.py
lgatr/primitives/__init__.py
lgatr/primitives/attention.py
lgatr/primitives/bilinear.py
lgatr/primitives/config.py
lgatr/primitives/dropout.py
lgatr/primitives/geometric_product.pt
lgatr/primitives/invariants.py
lgatr/primitives/linear.py
lgatr/primitives/linear_basis_full.pt
lgatr/primitives/linear_basis_subgroup.pt
lgatr/primitives/nonlinearities.py
lgatr/primitives/normalization.py
lgatr/primitives/attention_backends/__init__.py
lgatr/primitives/attention_backends/flash.py
lgatr/primitives/attention_backends/flex.py
lgatr/primitives/attention_backends/native.py
lgatr/primitives/attention_backends/varlen.py
lgatr/primitives/attention_backends/xformers.py
lgatr/utils/__init__.py
lgatr/utils/einsum.py
lgatr/utils/misc.py
tests/__init__.py
tests/helpers/__init__.py
tests/helpers/clifford.py
tests/helpers/constants.py
tests/helpers/equivariance.py
tests/helpers/equivariance_noga.py
tests/helpers/geometric_algebra.py
tests/lgatr/__init__.py
tests/lgatr/interface/__init__.py
tests/lgatr/interface/test_axialvector.py
tests/lgatr/interface/test_pseudoscalar.py
tests/lgatr/interface/test_scalar.py
tests/lgatr/interface/test_spurions.py
tests/lgatr/interface/test_vector.py
tests/lgatr/layers/__init__.py
tests/lgatr/layers/test_attention.py
tests/lgatr/layers/test_conditional_lgatr_block.py
tests/lgatr/layers/test_cross_attention.py
tests/lgatr/layers/test_dropout.py
tests/lgatr/layers/test_geometric_bilinears.py
tests/lgatr/layers/test_lgatr_block.py
tests/lgatr/layers/test_linear.py
tests/lgatr/layers/test_mlp.py
tests/lgatr/layers/test_nonlinearities.py
tests/lgatr/layers/test_normalization.py
tests/lgatr/nets/__init__.py
tests/lgatr/nets/test_conditional_lgatr.py
tests/lgatr/nets/test_conditional_lgatr_slim.py
tests/lgatr/nets/test_lgatr.py
tests/lgatr/nets/test_lgatr_slim.py
tests/lgatr/primitives/__init__.py
tests/lgatr/primitives/test_attention.py
tests/lgatr/primitives/test_attention_backends.py
tests/lgatr/primitives/test_bilinear.py
tests/lgatr/primitives/test_dropout.py
tests/lgatr/primitives/test_invariants.py
tests/lgatr/primitives/test_linear.py
tests/lgatr/primitives/test_nonlinearities.py
tests/lgatr/primitives/test_normalization.py
tests/lgatr/utils/__init__.py
tests/lgatr/utils/test_einsum.py
tests/lgatr/utils/test_misc.py