ricci 0.4.0

Graph neural network primitives: GCN and hyperbolic GCN on Burn tensors
Documentation 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22

//! ricci: graph learning primitives on Burn tensors.
//!
//! Provides Poincare ball geometry and graph convolution layers that run on any
//! Burn backend (ndarray, wgpu, tch, etc.).
//!
//! - [`PoincareBall`]: hyperbolic geometry ops (project, mobius_add, exp/log maps,
//!   hyperbolic activation via [`PoincareBall::hyp_act`])
//! - [`GCNConv`]: graph convolutional layer (linear + adjacency matmul)
//! - [`HGCNConv`]: hyperbolic graph convolution on the Poincare ball, with
//!   optional activation via [`HGCNConv::forward_act`]
//! - [`curvature`]: Ollivier-Ricci edge curvature (the crate's namesake), the
//!   primitive behind curvature-based rewiring of oversquashed graphs

#![forbid(unsafe_code)]

pub mod curvature;
pub mod hyperbolic;
pub mod nn;

pub use curvature::{ollivier_ricci_curvatures, CurvatureConfig, EdgeCurvature};
pub use hyperbolic::PoincareBall;
pub use nn::{GCNConv, HGCNConv};