diff --git a/jax/version.py b/jax/version.py
index <HASH>..<HASH> 100644
--- a/jax/version.py
+++ b/jax/version.py
@@ -12,4 +12,4 @@
 # See the License for the specific language governing permissions and
 # limitations under the License.
 
-__version__ = "0.1.52"
+__version__ = "0.1.53"
update version for pypi