oxicuda-gnn 0.2.0

Graph Neural Network primitives for OxiCUDA: sparse graph representations (CSR/COO/heterogeneous), message passing framework, GCN/GAT/GraphSAGE/GIN layers, global and hierarchical graph pooling — pure Rust, zero CUDA SDK dependency.
Documentation 
1
2
3
4
5
6
7
8

//! Graph readout functions.

pub mod dgi;
pub mod set2set;
pub mod sort_pool;

pub use dgi::{Dgi, DgiConfig, DgiLoss, DgiWeights};
pub use sort_pool::{SortPool, SortPoolConfig};