components

Components for constructing models.

Submodules

• attention_blocks
  • DropPath
  • Mlp
  • Attention_rel
  • Block_rel
  • Block
• cnn_convolutions
  • Conv3dBN
  • InceptionBlock4
  • InceptionResnet
• edge_convolutions
  • DynEdgeConv
  • EdgeConvTito
  • DynTrans
• embedding
  • LinearEdgeEncoder
  • LinearNodeEncoder
  • SinusoidalPosEmb
  • FourierEncoder
  • SpacetimeEncoder
  • RRWPLinearNodeEncoder
  • RRWPLinearEdgeEncoder
  • RWSELinearNodeEncoder
• 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()