Skip to main content

ruvector_attention/hyperbolic/
mod.rs

1//! Hyperbolic Attention Module
2//!
3//! Implements attention mechanisms in hyperbolic space using:
4//! - Poincaré ball model (traditional)
5//! - Lorentz hyperboloid model (novel - faster, more stable)
6
7pub mod hyperbolic_attention;
8pub mod lorentz_cascade;
9pub mod mixed_curvature;
10pub mod poincare;
11
12pub use poincare::{
13    exp_map, frechet_mean, log_map, mobius_add, mobius_scalar_mult, poincare_distance,
14    project_to_ball,
15};
16
17pub use hyperbolic_attention::{HyperbolicAttention, HyperbolicAttentionConfig};
18
19pub use mixed_curvature::{MixedCurvatureAttention, MixedCurvatureConfig};
20
21// Novel Lorentz Cascade Attention (LCA)
22pub use lorentz_cascade::{
23    busemann_score, einstein_midpoint, horosphere_attention_weights, lorentz_distance,
24    lorentz_inner, project_hyperboloid, CascadeHead, LCAConfig, LorentzCascadeAttention,
25};