Skip to content

KyleFromNVIDIA/wholegraph

 
 

Repository files navigation

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 content

About

WholeGraph - large scale Graph Neutral Networks

Resources

License

Stars

Watchers

Forks

Packages

No packages published

Languages

  • Cuda 41.1%
  • C++ 31.1%
  • Python 15.1%
  • Cython 5.2%
  • C 4.2%
  • CMake 1.8%
  • Other 1.5%