A TensorFlow library for building Graph Neural Networks with support for heterogeneous graphs and scalable data processing.
TensorFlow GNN is a library for building Graph Neural Networks (GNNs) within the TensorFlow ecosystem. It provides specialized tools for handling graph-structured data, including data representation, sampling, model building, and training orchestration. The library solves the problem of applying machine learning to relational data like social networks, molecular graphs, and knowledge graphs.
Machine learning engineers and researchers working with graph-structured data who need scalable, production-ready tools for building and training GNN models using TensorFlow.
Developers choose TensorFlow GNN because it offers a comprehensive, battle-tested solution for GNN development with support for heterogeneous graphs, scalable data processing, and seamless integration with TensorFlow's ecosystem. Its origin as a Google-internal library ensures robustness for real-world applications.
TensorFlow GNN is a library to build Graph Neural Networks on the TensorFlow platform.
Open-Awesome is built by the community, for the community. Submit a project, suggest an awesome list, or help improve the catalog on GitHub.
The GraphTensor type explicitly handles graphs with multiple node and edge types, as detailed in the schema guide, enabling complex relational data modeling.
Includes a graph sampler built with Apache Beam for converting large databases into manageable subgraphs, addressing challenges in training on massive graphs.
Offers high-level training orchestration APIs and pre-built models, ported from Google's internal library, ensuring robustness for real-world applications.
Seamlessly works with TensorFlow's tools and Keras layers, allowing developers to leverage existing ML workflows and infrastructure.
Does not support Keras v3, requiring manual setup like pip installing tf-keras and environment variables for TF2.16+, as admitted in the installation notes.
Developed and tested primarily on Linux, with potential compatibility issues on other platforms supported by TensorFlow, limiting cross-OS deployment.
Requires additional dependencies like Apache Beam for sampling and specific GPU drivers, increasing initial configuration complexity compared to lighter alternatives.