Index

C

ChangeOfFrames (class in lloca.framesnet.frames)

check_timelike_first() (in module lloca.utils.orthogonalize_4d)

D

device (lloca.framesnet.frames.Frames property)

dim (lloca.reps.tensorreps.TensorReps property) dtype (lloca.framesnet.frames.Frames property)

E

EquiEdgeConv (class in lloca.equivectors.mlp)

expand() (lloca.framesnet.frames.Frames method)

F

fix_init_weight() (lloca.backbone.particletransformer.ParticleTransformer method) forward() (lloca.backbone.attention.LLoCaAttention method) (lloca.backbone.graphnet.GraphNet method) (lloca.backbone.mlp.MLP method) (lloca.backbone.particlenet.ParticleNet method) (lloca.backbone.particletransformer.ParticleTransformer method) (lloca.backbone.transformer.Transformer method) (lloca.equivectors.lgatr.LGATrVectors method) (lloca.equivectors.mlp.EquiEdgeConv method) (lloca.equivectors.mlp.MLPVectors method) (lloca.equivectors.pelican.PELICANVectors method) (lloca.framesnet.equi_frames.LearnedPDFrames method) (lloca.framesnet.equi_frames.LearnedRestFrames method) (lloca.framesnet.equi_frames.LearnedSO13Frames method) (lloca.framesnet.equi_frames.LearnedSO2Frames method) (lloca.framesnet.equi_frames.LearnedSO3Frames method) (lloca.framesnet.equi_frames.LearnedZFrames method) (lloca.framesnet.nonequi_frames.IdentityFrames method) (lloca.framesnet.nonequi_frames.RandomFrames method) (lloca.reps.tensorreps_transform.TensorRepsTransform method)

Frames (class in lloca.framesnet.frames)

G

get_batch_from_ptr() (in module lloca.utils.utils) get_edge_attr() (in module lloca.utils.utils) get_edge_index_from_ptr() (in module lloca.utils.utils) get_edge_index_from_shape() (in module lloca.utils.utils) get_node_to_edge_ptr_fully_connected() (in module lloca.utils.utils)

get_nonlinearity() (in module lloca.equivectors.mlp) get_operation() (in module lloca.equivectors.mlp) get_ptr_from_batch() (in module lloca.utils.utils) get_trafo_type() (in module lloca.utils.rand_transforms) globalize_vecs_or_not() (lloca.framesnet.equi_frames.LearnedFrames method) GraphNet (class in lloca.backbone.graphnet)

I

IdentityFrames (class in lloca.framesnet.nonequi_frames) IndexSelectFrames (class in lloca.framesnet.frames) init_weights() (lloca.backbone.particletransformer.ParticleTransformer method)

init_weights_or_not() (lloca.framesnet.equi_frames.LearnedFrames method) InverseFrames (class in lloca.framesnet.frames) is_simplified() (lloca.reps.tensorreps.TensorReps method) is_sorted (lloca.reps.tensorreps.TensorReps property)

L

LearnedFrames (class in lloca.framesnet.equi_frames) LearnedPDFrames (class in lloca.framesnet.equi_frames) LearnedRestFrames (class in lloca.framesnet.equi_frames) LearnedSO13Frames (class in lloca.framesnet.equi_frames) LearnedSO2Frames (class in lloca.framesnet.equi_frames) LearnedSO3Frames (class in lloca.framesnet.equi_frames) LearnedZFrames (class in lloca.framesnet.equi_frames) LGATrVectors (class in lloca.equivectors.lgatr) lloca.equivectors.lgatr module lloca.equivectors.mlp module lloca.equivectors.pelican module lloca.utils.lorentz module lloca.utils.orthogonalize_3d module

lloca.utils.orthogonalize_4d module lloca.utils.polar_decomposition module lloca.utils.rand_transforms module lloca.utils.utils module LLoCaAttention (class in lloca.backbone.attention) LLoCaMessagePassing (class in lloca.backbone.lloca_message_passing) lorentz_cross() (in module lloca.utils.lorentz) lorentz_eye() (in module lloca.utils.lorentz) lorentz_inner() (in module lloca.utils.lorentz) lorentz_metric() (in module lloca.utils.lorentz) lorentz_squarednorm() (in module lloca.utils.lorentz) LowerIndicesFrames (class in lloca.framesnet.frames)

M

mass_regularize() (lloca.framesnet.equi_frames.LearnedFrames method) max_rep (lloca.reps.tensorreps.TensorReps property) message() (lloca.equivectors.lgatr.LGATrVectors method) (lloca.equivectors.mlp.EquiEdgeConv method) (lloca.equivectors.pelican.PELICANVectors method) metric (lloca.framesnet.frames.Frames property) MLP (class in lloca.backbone.mlp) MLPVectors (class in lloca.equivectors.mlp) module lloca.equivectors.lgatr lloca.equivectors.mlp lloca.equivectors.pelican lloca.utils.lorentz lloca.utils.orthogonalize_3d lloca.utils.orthogonalize_4d lloca.utils.polar_decomposition lloca.utils.rand_transforms lloca.utils.utils

mul (lloca.reps.tensorreps.TensorReps property) mul_scalars (lloca.reps.tensorreps.TensorReps property) mul_without_scalars (lloca.reps.tensorreps.TensorReps property)

N

no_weight_decay() (lloca.backbone.particletransformer.ParticleTransformer method)

normalize_4d() (in module lloca.utils.orthogonalize_4d)

O

orthogonalize_3d() (in module lloca.utils.orthogonalize_3d) orthogonalize_4d() (in module lloca.utils.orthogonalize_4d) orthogonalize_cross() (in module lloca.utils.orthogonalize_4d)

orthogonalize_cross_3d() (in module lloca.utils.orthogonalize_3d) orthogonalize_gramschmidt() (in module lloca.utils.orthogonalize_4d) orthogonalize_gramschmidt_3d() (in module lloca.utils.orthogonalize_3d) orthogonalize_wrapper_4d() (in module lloca.utils.orthogonalize_4d)

P

ParticleNet (class in lloca.backbone.particlenet) ParticleTransformer (class in lloca.backbone.particletransformer) PELICANVectors (class in lloca.equivectors.pelican)

polar_decomposition() (in module lloca.utils.polar_decomposition) pre_message_hook() (lloca.backbone.lloca_message_passing.LLoCaMessagePassing method) pre_propagate_hook() (lloca.backbone.lloca_message_passing.LLoCaMessagePassing method) prepare_frames() (lloca.backbone.attention.LLoCaAttention method)

R

rand_boost() (in module lloca.utils.rand_transforms) rand_lorentz() (in module lloca.utils.rand_transforms) rand_rotation() (in module lloca.utils.rand_transforms) rand_xyrotation() (in module lloca.utils.rand_transforms) rand_ztransform() (in module lloca.utils.rand_transforms) RandomFrames (class in lloca.framesnet.nonequi_frames)

regularize_collinear() (in module lloca.utils.orthogonalize_3d) regularize_coplanar() (in module lloca.utils.orthogonalize_4d) regularize_lightlike() (in module lloca.utils.orthogonalize_4d) repeat() (lloca.framesnet.frames.Frames method) reps (lloca.reps.tensorreps.TensorReps property) reshape() (lloca.framesnet.frames.Frames method) restframe_boost() (in module lloca.utils.polar_decomposition)

S

sample_rapidity() (in module lloca.utils.rand_transforms) shape (lloca.framesnet.frames.Frames property)

simplify() (lloca.reps.tensorreps.TensorReps method) softmax() (in module lloca.equivectors.mlp) sort() (lloca.reps.tensorreps.TensorReps method)

T

TensorReps (class in lloca.reps.tensorreps) TensorRepsTransform (class in lloca.reps.tensorreps_transform) to() (lloca.framesnet.frames.Frames method)

transform() (in module lloca.utils.rand_transforms) (lloca.framesnet.nonequi_frames.RandomFrames method) transform_parity() (lloca.reps.tensorreps_transform.TensorRepsTransform method) Transformer (class in lloca.backbone.transformer)