Skip to content

Useful extra functionality for TensorFlow 2.x maintained by SIG-addons

License

Notifications You must be signed in to change notification settings

bot-of-gabrieldemarmiesse/addons

 
 

Repository files navigation




PyPI Status Badge PyPI - Python Version Documentation Gitter chat Code style: black

Nightly Tests

Build Type Status
MacOS CPU Status
Windows CPU Status
Ubuntu CPU Status
Ubuntu GPU Status

TensorFlow Addons is a repository of contributions that conform to well-established API patterns, but implement new functionality not available in core TensorFlow. TensorFlow natively supports a large number of operators, layers, metrics, losses, and optimizers. However, in a fast moving field like ML, there are many interesting new developments that cannot be integrated into core TensorFlow (because their broad applicability is not yet clear, or it is mostly used by a smaller subset of the community).

Maintainers

Subpackage Maintainers Contact Info
tfa.activations SIG-Addons @facaiy @seanpmorgan
tfa.callbacks SIG-Addons @squadrick @shun-lin
tfa.image SIG-Addons @windqaq @facaiy
tfa.layers SIG-Addons @seanpmorgan @facaiy
tfa.losses SIG-Addons @facaiy @windqaq
tfa.metrics SIG-Addons @squadrick
tfa.optimizers SIG-Addons @facaiy @windqaq @squadrick
tfa.rnn Google @qlzh727
tfa.seq2seq Google/SIG-Addons @qlzh727 @guillaumekln
tfa.text SIG-Addons @seanpmorgan @facaiy

Installation

Stable Builds

TFA is available on PyPi for Linux/MacOS/Windows. To install the latest version, run the following:

pip install tensorflow-addons

To use addons:

import tensorflow as tf
import tensorflow_addons as tfa

Linux Build Matrix

Version Compatible With Python versions Compiler cuDNN CUDA
tfa-nightly tensorflow>=2.1.0 3.5-3.7 GCC 7.3.1 7.6 10.1
tensorflow-addons-0.8.2 tensorflow>=2.1.0 3.5-3.7 GCC 7.3.1 7.6 10.1
tensorflow-addons-0.7.1 tensorflow>=2.1.0 2.7, 3.5-3.7 GCC 7.3.1 7.6 10.1
tensorflow-addons-0.6.0 tensorflow==2.0.0 2.7, 3.5-3.7 GCC 7.3.1 7.4 10.0

Nightly Builds

There are also nightly builds of TensorFlow Addons under the pip package tfa-nightly, which is built against the latest stable version of TensorFlow. Nightly builds include newer features, but may be less stable than the versioned releases.

pip install tfa-nightly

Installing from Source

You can also install from source. This requires the Bazel build system (version >= 1.0.0).

git clone https://github.com/tensorflow/addons.git
cd addons

# This script links project with TensorFlow dependency
python3 ./configure.py

bazel build --enable_runfiles build_pip_pkg
bazel-bin/build_pip_pkg artifacts

pip install artifacts/tensorflow_addons-*.whl

Tutorials

See docs/tutorials/ for end-to-end examples of various addons.

Core Concepts

Standardized API within Subpackages

User experience and project maintainability are core concepts in TF-Addons. In order to achieve these we require that our additions conform to established API patterns seen in core TensorFlow.

GPU/CPU Custom-Ops

A major benefit of TensorFlow Addons is that there are precompiled ops. Should a CUDA 10.1 installation not be found then the op will automatically fall back to a CPU implementation.

Proxy Maintainership

Addons has been designed to compartmentalize subpackages and submodules so that they can be maintained by users who have expertise and a vested interest in that component.

Subpackage maintainership will only be granted after substantial contribution has been made in order to limit the number of users with write permission. Contributions can come in the form of issue closings, bug fixes, documentation, new code, or optimizing existing code. Submodule maintainership can be granted with a lower barrier for entry as this will not include write permissions to the repo.

For more information see the RFC on this topic.

Periodic Evaluation of Subpackages

Given the nature of this repository, subpackages and submodules may become less and less useful to the community as time goes on. In order to keep the repository sustainable, we'll be performing bi-annual reviews of our code to ensure everything still belongs within the repo. Contributing factors to this review will be:

  1. Number of active maintainers
  2. Amount of OSS use
  3. Amount of issues or bugs attributed to the code
  4. If a better solution is now available

Functionality within TensorFlow Addons can be categorized into three groups:

  • Suggested: well-maintained API; use is encouraged.
  • Discouraged: a better alternative is available; the API is kept for historic reasons; or the API requires maintenance and is the waiting period to be deprecated.
  • Deprecated: use at your own risk; subject to be deleted.

The status change between these three groups is: Suggested <-> Discouraged -> Deprecated.

The period between an API being marked as deprecated and being deleted will be 90 days. The rationale being:

  1. In the event that TensorFlow Addons releases monthly, there will be 2-3 releases before an API is deleted. The release notes could give user enough warning.

  2. 90 days gives maintainers ample time to fix their code.

Contributing

TF-Addons is a community led open source project. As such, the project depends on public contributions, bug-fixes, and documentation. Please see contribution guidelines for a guide on how to contribute. This project adheres to TensorFlow's code of conduct. By participating, you are expected to uphold this code.

Community

License

Apache License 2.0

About

Useful extra functionality for TensorFlow 2.x maintained by SIG-addons

Resources

License

Code of conduct

Stars

Watchers

Forks

Packages

No packages published

Languages

  • Python 75.0%
  • C++ 13.2%
  • Smarty 5.5%
  • Starlark 5.1%
  • Shell 1.0%
  • Dockerfile 0.2%