--- This fork includes support for NVIDIA Jetson AGX. Please use the r0.7 branch ---
| Build Type | Status |
|---|---|
| MacOS CPU |  |
| Windows CPU |  |
| Ubuntu CPU |  |
| Ubuntu GPU |  |
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).
| 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 | @qlzh727 | |
| tfa.seq2seq | Google/SIG-Addons | @qlzh727 @guillaumekln |
| tfa.text | SIG-Addons | @seanpmorgan @facaiy |
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| 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 |
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
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
See docs/tutorials/
for end-to-end examples of various addons.
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.
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.
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.
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:
- Number of active maintainers
- Amount of OSS use
- Amount of issues or bugs attributed to the code
- 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:
-
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.
-
90 days gives maintainers ample time to fix their code.
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.
- Public Mailing List
- SIG Monthly Meeting Notes
- Join our mailing list and receive calendar invites to the meeting