Expand description
Hyperbolic Attention Module
Implements attention mechanisms in hyperbolic space using:
- Poincaré ball model (traditional)
- Lorentz hyperboloid model (novel - faster, more stable)
Re-exports§
pub use poincare::exp_map;pub use poincare::frechet_mean;pub use poincare::log_map;pub use poincare::mobius_add;pub use poincare::mobius_scalar_mult;pub use poincare::poincare_distance;pub use poincare::project_to_ball;pub use hyperbolic_attention::HyperbolicAttention;pub use hyperbolic_attention::HyperbolicAttentionConfig;pub use mixed_curvature::MixedCurvatureAttention;pub use mixed_curvature::MixedCurvatureConfig;pub use lorentz_cascade::busemann_score;pub use lorentz_cascade::einstein_midpoint;pub use lorentz_cascade::horosphere_attention_weights;pub use lorentz_cascade::lorentz_distance;pub use lorentz_cascade::lorentz_inner;pub use lorentz_cascade::project_hyperboloid;pub use lorentz_cascade::CascadeHead;pub use lorentz_cascade::LCAConfig;pub use lorentz_cascade::LorentzCascadeAttention;
Modules§
- hyperbolic_
attention - Hyperbolic Attention Mechanism using Poincaré ball model
- lorentz_
cascade - Lorentz Cascade Attention (LCA) - A Novel Hyperbolic Attention Mechanism
- mixed_
curvature - Mixed-Curvature Attention combining Euclidean and Hyperbolic spaces
- poincare
- Poincaré Ball Model Operations for Hyperbolic Geometry