Skip to main content

Graph engine - distributed graph engine to host graphs.

Project description

DeepGNN Overview

DeepGNN is a framework for training machine learning models on large scale graph data. DeepGNN contains all the necessary features including:

  • Distributed GNN training and inferencing on both CPU and GPU.
  • Custom graph neural network design.
  • Online Sampling: Graph Engine (GE) will load all graph data, each training worker will call GE to get node/edge/neighbor features and labels.
  • Automatic graph partitioning.
  • Highly performant and scalable.

Project is in alpha version, there might be breaking changes in the future and they will be documented in the changelog.


Install pip package:

python -m pip install deepgnn

If you want to build package from source, see instructions in

Train and evaluate a graphsage model with pytorch on cora dataset:

cd examples/pytorch

Migrating Scripts

We provide a python module to help you upgrade your scripts to new deepgnn versions.

pip install google-pasta
python -m deepgnn.migrate.0_1_56 --script_dir directory_to_migrate

See for full change details.

Project details

Download files

Download the file for your platform. If you're not sure which to choose, learn more about installing packages.

Source Distributions

No source distribution files available for this release.See tutorial on generating distribution archives.

Built Distributions

deepgnn_ge-0.1.63-py3-none-win_amd64.whl (3.1 MB view hashes)

Uploaded Python 3 Windows x86-64

deepgnn_ge-0.1.63-py3-none-manylinux1_x86_64.whl (11.9 MB view hashes)

Uploaded Python 3

deepgnn_ge-0.1.63-py3-none-macosx_10_9_x86_64.whl (4.6 MB view hashes)

Uploaded Python 3 macOS 10.9+ x86-64

Supported by

AWS AWS Cloud computing and Security Sponsor Datadog Datadog Monitoring Fastly Fastly CDN Google Google Download Analytics Microsoft Microsoft PSF Sponsor Pingdom Pingdom Monitoring Sentry Sentry Error logging StatusPage StatusPage Status page