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