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 pooling operations.

pub mod diff_pool;
pub mod global_pool;
pub mod sag_pool;
pub mod topk_pool;

pub use sag_pool::{SagPool, SagPoolResult};