For a library for graph neural networks, the first results are rusty1s/pytorch_geometric, dmlc/dgl (DGL is a graph neural network library with native support for heterogeneous graphs and relational convolution operators, making it exactly the kind of tool for building GNNs on multi-relational knowledge graphs and other relational data) and pyg-team/pytorch_geometric. pytorch/pytorch is also worth a look. Compare the match explanations and check the project documentation against your requirements.
Frameworks and tools for implementing graph neural networks specifically designed to process complex relational data structures.
PyTorch Geometric is a library for building and training machine learning models on graph-structured data. It provides a framework for developing graph neural networks, including a specialized system for implementing node-to-node information exchange via customizable message passing, aggregation, and update functions. The library includes a sparse data processing toolkit that utilizes accelerated CPU and CUDA kernels to perform efficient reductions on large sparse datasets. It supports the creation of specialized architectures for structured data such as 3D meshes and point clouds. The proje
PyTorch Geometric is a comprehensive graph neural network library that fully supports heterogeneous graphs and relational graph convolution operators (e.g., RGCN, HGT) within PyTorch, providing built-in datasets, scalability, and extensive tutorials for relational GNNs, making it an exact fit for your search.
DGL is a Python library for building and training graph neural networks. It functions as a graph message passing framework and a geometric deep learning tool, enabling the development of models that analyze graph-structured data. The library is designed for large-scale graph processing, utilizing distributed training and neighbor sampling to handle datasets with billions of edges. It provides specialized support for heterogeneous graph modeling, allowing for the representation of complex real-world entities with multiple node and edge types. Its capabilities cover a wide range of graph tasks
DGL is a graph neural network library with native support for heterogeneous graphs and relational convolution operators, making it exactly the kind of tool for building GNNs on multi-relational knowledge graphs and other relational data.
This project is a deep learning library designed for training neural networks on irregular data structures, including graphs, 3D meshes, and point clouds. It functions as an extension to the PyTorch framework, providing specialized layers and kernels that enable the processing of complex, non-Euclidean information. The library distinguishes itself through a geometric deep learning toolkit that manages the unique requirements of graph-based data. It utilizes sparse matrix-based message passing to aggregate information across nodes and employs dynamic computational graph construction to accommo
PyTorch Geometric is a dedicated graph neural network library built on PyTorch that natively supports heterogeneous graphs and multi-relational data through specialized data structures and operators like RGCN, making it a comprehensive and flagship choice for relational GNN research and applications.
PyTorch is a machine learning framework centered on a GPU-ready tensor library that supports multi-dimensional array operations across both CPU and accelerator hardware. It provides a foundational infrastructure for mathematical computation and dynamic neural network construction, utilizing a tape-based automatic differentiation system that allows for flexible, non-static graph execution. The framework is designed for deep integration with Python, enabling natural usage alongside standard scientific computing ecosystems. It distinguishes itself through a comprehensive distributed training sui
PyTorch is a general-purpose deep learning framework, not a graph neural network library specialized for relational or heterogeneous data — you would use PyTorch as a foundation to build such a library, but it lacks built-in support for heterogeneous graphs and relational graph convolution operators.
| Repository | Stars | Language | License | Last push |
|---|---|---|---|---|
| rusty1s/pytorch_geometric | 23.8K | Python | MIT | |
| dmlc/dgl | 14.3K | Python | Apache-2.0 | |
| pyg-team/pytorch_geometric | 23.8K | Python | MIT | |
| pytorch/pytorch | 100.8K | Python | NOASSERTION |