All modules for which code is available
- lgatr.utils.autocast
- lloca.backbone.attention
- lloca.backbone.attention_backends
- lloca.backbone.graphnet
- lloca.backbone.lloca_message_passing
- lloca.backbone.mlp
- lloca.backbone.particlenet
- lloca.backbone.particletransformer
- lloca.backbone.transformer
- lloca.backbone.transformer_v2
- lloca.equivectors.lgatr
- lloca.equivectors.mlp
- lloca.equivectors.pelican
- lloca.framesnet.equi_frames
- lloca.framesnet.frames
- lloca.framesnet.nonequi_frames
- lloca.reps.tensorreps
- lloca.reps.tensorreps_transform
- lloca.utils.compile
- lloca.utils.lorentz
- lloca.utils.orthogonalize_3d
- lloca.utils.orthogonalize_4d
- lloca.utils.polar_decomposition
- lloca.utils.rand_transforms
- lloca.utils.utils