diff --git a/CHANGELOG.md b/CHANGELOG.md
index f205847..bbc89bf 100644
--- a/CHANGELOG.md
+++ b/CHANGELOG.md
@@ -1,17 +1,21 @@
 # Changelog
 
-## [unreleased][unreleased]
+## [0.5.2][2021-07-15]
 
 ### Fixed
 
 -   Fixed error in default arguments to `tfrs.experimental.models.Ranking`
     (https://github.com/tensorflow/recommenders/issues/311).
+-   Fix TPUEmbedding layer to use named parameters.
 
 ### Added
 
 -   Added `batch_metrics` to `tfrs.tasks.Retrieval` for measuring how good the
     model is at picking out the true candidate for a query from other candidates
     in the batch.
+-   Added `tfrs.experimental.layers.embedding.PartialTPUEmbedding` layer, which
+    uses `tfrs.layers.embedding.TPUEmbedding` for large embedding lookups and
+    `tf.keras.layers.Embedding` for smaller embedding lookups.
 
 ## [0.5.1][2021-05-14]
 
diff --git a/setup.py b/setup.py
index e069d35..7c67d09 100644
--- a/setup.py
+++ b/setup.py
@@ -17,7 +17,7 @@
 import pathlib
 import setuptools
 
-VERSION = "0.5.1"
+VERSION = "0.5.2"
 
 long_description = (pathlib.Path(__file__).parent
                     .joinpath("README.md")
diff --git a/tensorflow_recommenders/__init__.py b/tensorflow_recommenders/__init__.py
index 8c41b57..1f21065 100644
--- a/tensorflow_recommenders/__init__.py
+++ b/tensorflow_recommenders/__init__.py
@@ -21,7 +21,7 @@
 you the flexibility to build complex models.
 """
 
-__version__ = "v0.5.1"
+__version__ = "v0.5.2"
 
 from tensorflow_recommenders import examples
 from tensorflow_recommenders import experimental