Graph Neural Network Frameworks

Graph neural network (GNN) frameworks are easy-to-use Python packages that offer building blocks to build GNNs on top of existing deep learning frameworks for a wide range of applications.

Get started

Graph Neural Network Frameworks

NVIDIA AI Accelerated GNN frameworks are optimized to deliver high-performance preprocessing, sampling, and training on NVIDIA GPUs.

Explore the benefits.

GNN framework containers for DGL and PyG

NVIDIA AI optimized GNN frameworks.

GNN framework containers for Deep Graph Library (DGL) and PyTorch Geometric (PyG) come with the latest NVIDIA RAPIDs, PyTorch, and frameworks that are performance tuned and tested for NVIDIA GPUs.

GPU-accelerated ETL

GPU-accelerated ETL.

Go from hours to minutes. With NVIDIA RAPIDS? integration, cuDF accelerates pandas queries up to 39X faster than CPU so that you can run ETL with GPU-optimized code.

End-to-end reference examples

End-to-end reference examples.

Streamline workflows for GNNs, from experimentation to production, with GPU-optimized, tested, and validated examples for fraud detection, recommender systems, and drug discovery.

Runs interference on GNNs in the cloud

Runs on prem and in cloud.

Preprocess, train, and run inference on graph neural networks on multi-GPU, multi-node infrastructure.

See performance benchmarks.

With NVIDIA AI Accelerated GNN frameworks, you can get end-to-end performance optimization, making it the fastest solution to preprocess and build GNNs.

DGL Container, Dataset: MAG240M, Model: RCGN, Total edges: 1.7B
GPU: 1x A100 80GB, CPU: AMD EPYC 7742 64-Core

NVIDIA AI Accelerated GNN frameworks.

Deep Graph Library

Deep Graph Library (DGL) is an easy-to-use and scalable Python library used for implementing and training GNNs.

To enable developers to quickly take advantage of GNNs, we’ve partnered with the DGL team to provide a containerized solution that includes the latest DGL, PyTorch, and NVIDIA RAPIDS (cuDF, XGBoost, RMM, cuML, and cuGraph), which can be used to accelerate ETL operations and training.

Our private early access release includes two containers:

  1. A ready-to-use DGL container with the latest upstream improvements and tested dependencies.
  2. A ready-to-use DGL container with tested dependencies, an optimized SE(3)-Transformer model, and an accelerated neural network training environment based on DGL and PyTorch. The SE(3)-Transformer for DGL container is suited for recognizing three-dimensional shapes making it useful for segmenting lidar point clouds or in pharmaceutical and drug discovery research.

Apply for early access to our DGL container or the SE(3)-Transformer for DGL container.

Apply for Early Access
Graph Neural Network Frameworks
Graph Neural Network Frameworks

PyTorch Geometric

PyTorch Geometric (PyG) is a library built upon PyTorch to easily write and train GNNs for a wide range of applications related to structured data.

We've collaborated with the PyG team to offer the best performance on NVIDIA GPUs—a containerized solution that includes the latest version of PyG, PyTorch, and NVIDIA RAPIDS, which can be used to speed up the GNN workflow on GPUs.

A ready-to-use PyG container with the latest upstream improvements and tested dependencies will be available in Q4’2022 in private early access.

Apply for Early Access

How GNNs are being used across industries

Fraud detection in financial services

Fraud detection in financial services.

Fraudulent transactions are some of the most serious threats to the financial services industry. GNNs have emerged as a powerful tool to improve fraud detection tasks where fraudulent transactions are identified by aggregating neighbor information.

Drug discovery in healthcare.

Drug discovery in healthcare.

To develop new candidate medications quickly, researchers need to compute the physical properties of molecules. GNNs provide an efficient way to represent these complex 3D structures and accurately predict their properties.

Recommenders in retail.

Recommenders in retail.

GNNs can leverage information from existing data with a graph structure, resulting in better predictions than traditional methods. Recommendation systems use a form of node embeddings in GNNs to match customers with products.

Get started.

Get started with DGL.

Check out tutorials on how to get started with the NVIDIA DGL Container and walk through examples of preprocessing, training, and inference.

Learn more

Watch GTC sessions.

Learn more about GNN frameworks and their new features from a curated list of webinars from GTC.

Watch sessions

What are GNNs?

Learn about GNNs with this gentle introduction.

Explore blog