propago 0.2.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

//! propago: 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`]

#![forbid(unsafe_code)]

pub mod hyperbolic;
pub mod nn;

pub use hyperbolic::PoincareBall;
pub use nn::{GCNConv, HGCNConv};