Attend the Women in ML Symposium on December 7 Register now

Module: tfa

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

Useful extra functionality for TensorFlow maintained by SIG-addons.


activations module: Additional activation functions.

callbacks module: Additional callbacks that conform to Keras API.

image module: Additional image manipulation ops.

layers module: Additional layers that conform to Keras API.

losses module: Additional losses that conform to Keras API.

metrics module: Additional metrics that conform to Keras API.

optimizers module: Additional optimizers that conform to Keras API.

options module

rnn module: Additional RNN cells that corform to Keras API.

seq2seq module: Additional layers for sequence to sequence models.

text module: Additional text-processing ops.

types module: Types for typing functions signatures.


register_all(...): Register TensorFlow Addons' objects in TensorFlow global dictionaries.