diff --git a/drjax/__init__.py b/drjax/__init__.py
index 1cecb25..fce101a 100644
--- a/drjax/__init__.py
+++ b/drjax/__init__.py
@@ -19,7 +19,7 @@
 
 from drjax._src import api as _api
 
-__version__ = '0.1.3'
+__version__ = '0.1.4'
 
 # Import the public API.
 broadcast = _api.broadcast