rapidsai / wholegraph

WholeGraph - large scale Graph Neural Networks

Home Page:https://docs.rapids.ai/api/cugraph/stable/wholegraph/

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

WholeGraph

WholeGraph is developed to help train large-scale Graph Neural Networks(GNN). WholeGraph provides underlying storage structure called WholeMemory. WholeMemory is a Tensor like storage and provide multi-GPU support. It is optimized for NVLink systems like DGX A100 servers. By working together with cuGraph, cuGraph-Ops, cuGraph-DGL, cuGraph-PyG, and upstream DGL and PyG, it will be easy to build GNN applications.

Table of Contents

About

WholeGraph - large scale Graph Neural Networks

https://docs.rapids.ai/api/cugraph/stable/wholegraph/

License:Apache License 2.0


Languages

Language:Cuda 40.8%Language:C++ 33.6%Language:Python 13.6%Language:Cython 4.9%Language:C 4.0%Language:CMake 1.6%Language:Shell 1.2%Language:HTML 0.1%Language:Dockerfile 0.0%