diff --git a/CHANGELOG.md b/CHANGELOG.md
index 4464721d..e4bc7998 100644
--- a/CHANGELOG.md
+++ b/CHANGELOG.md
@@ -7,6 +7,8 @@ and this project adheres to [Semantic Versioning](https://semver.org/spec/v2.0.0
 
 ## [Unreleased]
 
+## [0.1.7] - 2022-03-29
+
 ### Added
 - Support for OCDBT driver in Tensorstore.
 
diff --git a/orbax/__init__.py b/orbax/__init__.py
index 942946c8..5517f51a 100644
--- a/orbax/__init__.py
+++ b/orbax/__init__.py
@@ -15,4 +15,4 @@
 """Orbax API."""
 
 # A new PyPI release will be pushed everytime `__version__` is increased.
-__version__ = '0.1.6'
+__version__ = '0.1.7'
diff --git a/orbax/checkpoint/__init__.py b/orbax/checkpoint/__init__.py
index 071fc088..3a41b510 100644
--- a/orbax/checkpoint/__init__.py
+++ b/orbax/checkpoint/__init__.py
@@ -19,6 +19,7 @@
 from orbax.checkpoint import aggregate_handlers
 from orbax.checkpoint import checkpoint_utils
 from orbax.checkpoint import lazy_utils
+from orbax.checkpoint import msgpack_utils
 from orbax.checkpoint import test_utils
 from orbax.checkpoint import type_handlers
 from orbax.checkpoint import utils