components

Components for constructing models.

Submodules

• embedding
  • LinearEdgeEncoder
  • LinearNodeEncoder
  • SinusoidalPosEmb
  • FourierEncoder
  • SpacetimeEncoder
  • RRWPLinearNodeEncoder
  • RRWPLinearEdgeEncoder
  • RWSELinearNodeEncoder
• layers
  • DynEdgeConv
  • EdgeConvTito
  • DynTrans
  • DropPath
  • Mlp
  • Block_rel
  • Attention_rel
  • Block
  • GritSparseMHA
  • GritTransformerLayer
  • SANGraphHead
• 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()