diff --git a/README.md b/README.md
index 23eb4b35..c14e6ef9 100644
--- a/README.md
+++ b/README.md
@@ -199,7 +199,7 @@ To cite this project:
 author = {PoetsAI},
 title = {Elegy: A framework-agnostic Trainer interface for the Jax ecosystem},
 url = {https://github.com/poets-ai/elegy},
-version = {0.7.1},
+version = {0.7.2},
 year = {2020},
 }
 ```
diff --git a/docs/index.md b/docs/index.md
index 23eb4b35..c14e6ef9 100644
--- a/docs/index.md
+++ b/docs/index.md
@@ -199,7 +199,7 @@ To cite this project:
 author = {PoetsAI},
 title = {Elegy: A framework-agnostic Trainer interface for the Jax ecosystem},
 url = {https://github.com/poets-ai/elegy},
-version = {0.7.1},
+version = {0.7.2},
 year = {2020},
 }
 ```
diff --git a/elegy/__init__.py b/elegy/__init__.py
index 06efef27..011baea7 100644
--- a/elegy/__init__.py
+++ b/elegy/__init__.py
@@ -1,4 +1,4 @@
-__version__ = "0.7.1"
+__version__ = "0.7.2"
 
 
 from elegy import types
diff --git a/pyproject.toml b/pyproject.toml
index a651d2ea..bf5986ac 100644
--- a/pyproject.toml
+++ b/pyproject.toml
@@ -5,7 +5,7 @@ authors = ["Cristian Garcia <cgarcia.e88@gmail.com>",
            "Carlos Alvarez <candres.alv@gmail.com>",
            "David Cardozo <davidcardozo@berkeley.edu>",
            "Sebastian Arango"]
-version = "0.7.1"
+version = "0.7.2"
 license = "APACHE"
 readme = "README.md"
 repository = "https://github.com/poets-ai/elegy"