CUDA-X Accelerated DGL Containers for Large Graph Neural Networks

Originally published at: https://developer.nvidia.com/blog/cuda-x-accelerated-dgl-containers-for-large-graph-neural-networks/

NVIDIA partnered with the DGL team to provide containers with the latest DGL, PyTorch, and SE(3)-Transformer for GPU-accelerated performance optimization.

It was fun collaborating with the team in creating this post. I’m looking forward to seeing DGL containers being downloaded and in great use.