WARNING:
Depending on tensorflow_addons will transitively require TFF to pull in TF 2.0
pip packages; hence this is backwards incompatible with previous TFF versions
which were compatible with TF 1.x.
PiperOrigin-RevId: 272744011