CUDA-X Accelerated DGL Containers for Large Graph Neural Networks

Originally published at: CUDA-X Accelerated DGL Containers for Large Graph Neural Networks | NVIDIA Developer Blog

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.