pytorch-geometric

Features and purpose

Library for Graph Neural Networks (GNNs). Covers MessagePassing layers, modular aggregation schemes, and handling large graphs via mini-batching with disjoint graph representation. (pyg, messagepassing, gnn, gcn, gat, edge_index, knn_graph, global_mean_pool)

Quick start

Follow the repository instructions for this skill and your agent. Review the files and dependencies before adding it to your workflow.