Module graph

Module graph 

Source
Expand description

Graph attention mechanisms for GNN applications

This module provides graph-specific attention implementations:

  • Edge-featured attention (GAT with edge features)
  • Rotary position embeddings for graphs (RoPE)
  • Dual-space attention (Euclidean + Hyperbolic)

Re-exports§

pub use dual_space::DualSpaceAttention;
pub use dual_space::DualSpaceConfig;
pub use edge_featured::EdgeFeaturedAttention;
pub use edge_featured::EdgeFeaturedConfig;
pub use rope::GraphRoPE;
pub use rope::RoPEConfig;

Modules§

dual_space
Dual-space attention combining Euclidean and Hyperbolic geometries
edge_featured
Edge-featured graph attention (GATv2 style)
rope
Rotary Position Embeddings (RoPE) for Graph Attention