Skip to main content

Crate propago

Crate propago 

Source
Expand description

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.).

Re-exports§

pub use hyperbolic::PoincareBall;
pub use nn::GCNConv;
pub use nn::HGCNConv;

Modules§

hyperbolic
Poincare ball geometry on Burn tensors.
nn
Graph neural network layers on Burn tensors.