LICENSE
README.md
pyproject.toml
lgatr/__init__.py
lgatr.egg-info/PKG-INFO
lgatr.egg-info/SOURCES.txt
lgatr.egg-info/dependency_links.txt
lgatr.egg-info/requires.txt
lgatr.egg-info/top_level.txt
lgatr/interface/__init__.py
lgatr/interface/scalar.py
lgatr/interface/spurions.py
lgatr/interface/vector.py
lgatr/layers/__init__.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/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/lgatr.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/utils/__init__.py
lgatr/utils/einsum.py
lgatr/utils/misc.py