componentsΒΆ Components for constructing models. Submodules attention_blocks DropPath Mlp Attention_rel Block_rel Block SwiGLU AttentionPool RotaryTransformerBlock cnn_convolutions Conv3dBN InceptionBlock4 InceptionResnet edge_convolutions DynEdgeConv EdgeConvTito DynTrans embedding LinearEdgeEncoder LinearNodeEncoder SinusoidalPosEmb FourierEncoderEPJC FourierEncoder RoPE4D signed_four_distance() SpacetimeEncoderEPJC SpacetimeEncoder RRWPLinearNodeEncoder RRWPLinearEdgeEncoder RWSELinearNodeEncoder fps farthest_point_sampling() farthest_point_sampling_with_knn() farthest_point_sampling_with_assign() nearest_assign() grit_layers GritSparseMHA GritTransformerLayer SANGraphHead layers pool min_pool() min_pool_x() sum_pool_and_distribute() group_by() group_pulses_to_dom() group_pulses_to_pmt() sum_pool_x() std_pool_x() sum_pool() std_pool() tokenizers FPSTokenizer