diff --git a/tensorflow_addons/__init__.py b/tensorflow_addons/__init__.py
index 2afa5c13ad..b08f82bced 100644
--- a/tensorflow_addons/__init__.py
+++ b/tensorflow_addons/__init__.py
@@ -29,5 +29,6 @@
 from tensorflow_addons import seq2seq
 from tensorflow_addons import text
 from tensorflow_addons.register import register_all
+from tensorflow_addons.utils import types
 
 from tensorflow_addons.version import __version__
diff --git a/tensorflow_addons/image/__init__.py b/tensorflow_addons/image/__init__.py
index 5351a26a30..5781f8534d 100644
--- a/tensorflow_addons/image/__init__.py
+++ b/tensorflow_addons/image/__init__.py
@@ -34,6 +34,9 @@
 from tensorflow_addons.image.transform_ops import shear_x
 from tensorflow_addons.image.transform_ops import shear_y
 from tensorflow_addons.image.sparse_image_warp import sparse_image_warp
+from tensorflow_addons.image.transform_ops import compose_transforms
+from tensorflow_addons.image.transform_ops import angles_to_projective_transforms
 from tensorflow_addons.image.transform_ops import transform
 from tensorflow_addons.image.translate_ops import translate
 from tensorflow_addons.image.translate_ops import translate_xy
+from tensorflow_addons.image.translate_ops import translations_to_projective_transforms
diff --git a/tensorflow_addons/metrics/__init__.py b/tensorflow_addons/metrics/__init__.py
index 16df926649..8087413d49 100755
--- a/tensorflow_addons/metrics/__init__.py
+++ b/tensorflow_addons/metrics/__init__.py
@@ -16,7 +16,11 @@
 
 from tensorflow_addons.metrics.cohens_kappa import CohenKappa
 from tensorflow_addons.metrics.f_scores import F1Score, FBetaScore
-from tensorflow_addons.metrics.hamming import HammingLoss, hamming_distance
+from tensorflow_addons.metrics.hamming import (
+    HammingLoss,
+    hamming_distance,
+    hamming_loss_fn,
+)
 from tensorflow_addons.metrics.utils import MeanMetricWrapper
 from tensorflow_addons.metrics.matthews_correlation_coefficient import (
     MatthewsCorrelationCoefficient,
diff --git a/tensorflow_addons/optimizers/__init__.py b/tensorflow_addons/optimizers/__init__.py
index a565533632..5d03b6f3ca 100644
--- a/tensorflow_addons/optimizers/__init__.py
+++ b/tensorflow_addons/optimizers/__init__.py
@@ -42,4 +42,7 @@
 from tensorflow_addons.optimizers.weight_decay_optimizers import (
     extend_with_decoupled_weight_decay,
 )
+from tensorflow_addons.optimizers.weight_decay_optimizers import (
+    DecoupledWeightDecayExtension,
+)
 from tensorflow_addons.optimizers.yogi import Yogi
diff --git a/tensorflow_addons/text/__init__.py b/tensorflow_addons/text/__init__.py
index dfb6379ec9..7f3a890658 100644
--- a/tensorflow_addons/text/__init__.py
+++ b/tensorflow_addons/text/__init__.py
@@ -15,6 +15,8 @@
 """Additional text-processing ops."""
 
 # Conditional Random Field
+from tensorflow_addons.text import crf
+from tensorflow_addons.text.crf import CrfDecodeForwardRnnCell
 from tensorflow_addons.text.crf import crf_binary_score
 from tensorflow_addons.text.crf import crf_constrained_decode
 from tensorflow_addons.text.crf import crf_decode
diff --git a/tools/docs/build_docs.py b/tools/docs/build_docs.py
index 4e55502915..5442a42d52 100644
--- a/tools/docs/build_docs.py
+++ b/tools/docs/build_docs.py
@@ -74,8 +74,11 @@ def main(argv):
         py_modules=[(PROJECT_SHORT_NAME, tfa)],
         code_url_prefix=code_url_prefix,
         private_map={"tfa": ["__version__", "utils", "version"]},
-        # This callback usually cleans up a lot of aliases caused by internal imports.
-        callbacks=[public_api.local_definitions_filter],
+        # These callbacks usually clean up a lot of aliases caused by internal imports.
+        callbacks=[
+            public_api.local_definitions_filter,
+            public_api.explicit_package_contents_filter,
+        ],
         search_hints=FLAGS.search_hints,
         site_path=FLAGS.site_path,
     )
diff --git a/tools/install_deps/doc_requirements.txt b/tools/install_deps/doc_requirements.txt
index 0d162713fa..7a7a9edb11 100644
--- a/tools/install_deps/doc_requirements.txt
+++ b/tools/install_deps/doc_requirements.txt
@@ -1,2 +1,2 @@
-git+https://github.com/tensorflow/docs@8192df1
+git+https://github.com/tensorflow/docs
 pyyaml
\ No newline at end of file