diff --git a/tf2jax/__init__.py b/tf2jax/__init__.py
index f8f78ca..7e910e2 100644
--- a/tf2jax/__init__.py
+++ b/tf2jax/__init__.py
@@ -27,7 +27,7 @@
 
 from tf2jax._src.tf2jax import MissingInputError
 
-__version__ = "0.3.5"
+__version__ = "0.3.6"
 
 #  _________________________________________
 # / Please don't use symbols in `_src` they \