Stay organized with collections Save and categorize content based on your preferences.

Libraries & extensions

Explore libraries to build advanced models or methods using TensorFlow, and access domain-specific application packages that extend TensorFlow.

  • TensorFlow Addons

    Extra functionality for TensorFlow, maintained by SIG Addons.
    View docs
  • TensorFlow Agents

    A library for designing, testing, and implementing reinforcement learning algorithms.
    View docs
  • TensorFlow Compression

    A library to build ML models with end-to-end optimized data compression built in.
    View docs
  • TensorFlow Data Validation

    A library to analyze training and serving data to compute descriptive statistics, infer schemas, and detect anomalies.
    View docs
  • TensorFlow Decision Forests

    State-of-the-art algorithms for training, serving and interpreting models that use decision forests for classification, regression and ranking.
    View docs
  • Dopamine

    A research framework for fast prototyping of reinforcement learning algorithms.
  • Fairness Indicators

    A library that enables easy computation of commonly-identified fairness metrics for binary and multiclass classifiers.
    View docs
  • TensorFlow Federated

    An open source framework for machine learning and other computations on decentralized data.
    View docs
  • TensorFlow GNN

    A library to build neural networks on graph data (nodes and edges with arbitrary features), including tools for preparing input data and training models.
  • TensorFlow Graphics

    A library of computer graphics functionalities ranging from cameras, lights, and materials to renderers.
    View docs
  • TensorFlow Hub

    A library for reusable machine learning. Download and reuse the latest trained models with a minimal amount of code.
    View docs
  • TensorFlow IO

    Dataset, streaming, and file system extensions, maintained by SIG IO.
    View docs
  • TensorFlow JVM

    Language bindings for Java and other JVM languages, such as Scala or Kotlin.
    View docs
  • KerasCV

    A library of modular components for common computer vision tasks such as data augmentation, classification, object detection, segmentation, and more.
  • KerasNLP

    An easily customizable natural language processing library providing modular components and state-of-the-art preset weights and architectures.
  • TensorFlow Lattice

    A library for flexible, controlled and interpretable ML solutions with common-sense shape constraints.
    View docs
  • TensorFlow Lite Micro

    A library to run ML models on digital signal processors (DSPs), microcontrollers, and other devices with limited memory.
    View docs
  • TensorFlow Lite Model Maker

    A library that simplifies model training for on-device natural language processing, vision, and audio applications.
    View docs
  • TensorFlow Lite Support

    A toolkit to customize model interface on Android, create metadata, and build inference pipelines for mobile deployment.
    View docs
  • Utilities for passing TensorFlow-related metadata between tools.
  • A library for recording and retrieving MLOps metadata associated with machine learning workflows.
    View docs
  • TensorFlow Model Analysis

    A library for deep analysis of model results beyond simple training metrics, to measure edge and corner cases and bias.
    View docs
  • A collection of tools to generate documents that provide context and transparency into a model's development and performance.
    View docs
  • A suite of tools for optimizing ML models for deployment and execution.
    View docs
  • A library to help create and train models in a way that reduces or eliminates user harm resulting from underlying performance biases.
    View docs
  • NdArray

    Utilities for manipulating data in a n-dimensional space in Java, maintained by SIG JVM.
  • Neural Structured Learning

    A learning framework to train neural networks by leveraging structured signals in addition to feature inputs.
    View docs
  • TensorFlow Privacy

    A Python library that includes implementations of TensorFlow optimizers for training machine learning models with differential privacy.
    View docs
  • TensorFlow Probability

    A library for probabilistic reasoning and statistical analysis.
    View docs
  • TensorFlow Quantum

    A quantum machine learning library for rapid prototyping of hybrid quantum-classical ML models.
    View docs
  • TensorFlow Ranking

    A library for Learning-to-Rank (LTR) techniques on the TensorFlow platform.
    View docs
  • TensorFlow Recommenders

    A library for building recommender system models.
    View docs
  • TensorFlow Recommenders Addons

    A collection of community projects introducing Dynamic Embedding Technology to large-scale recommendation systems built upon TensorFlow
  • TensorFlow Serving

    A flexible, high-performance serving system for machine learning models, designed for production environments
    View docs
  • Sonnet

    A library from DeepMind for constructing neural networks.
  • TensorFlow Text

    A collection of text- and NLP-related classes and ops ready to use with TensorFlow 2.
    View docs
  • TensorFlow Transform

    A library for large-scale feature engineering and eliminating training-serving skew.
    View docs
  • TensorFlow.js

    A hardware-accelerated library for training and deploying ML models using JavaScript or Node.js.
    View docs
  • TFX

    An end-to-end platform for deploying production ML pipelines.
    View docs
  • TFX-Addons

    A collection of community projects to build new components, examples, libraries, and tools for TFX.
    View docs

[[["Easy to understand","easyToUnderstand","thumb-up"],["Solved my problem","solvedMyProblem","thumb-up"],["Other","otherUp","thumb-up"]],[["Missing the information I need","missingTheInformationINeed","thumb-down"],["Too complicated / too many steps","tooComplicatedTooManySteps","thumb-down"],["Out of date","outOfDate","thumb-down"],["Samples / code issue","samplesCodeIssue","thumb-down"],["Other","otherDown","thumb-down"]],[],[],[]]