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