diff --git a/zuko/__init__.py b/zuko/__init__.py
index f078a9d..66ae3f5 100644
--- a/zuko/__init__.py
+++ b/zuko/__init__.py
@@ -1,6 +1,6 @@
 r"""Zuko - Normalizing flows in PyTorch"""
 
-__version__ = '0.1.5'
+__version__ = '0.2.0'
 
 from . import distributions
 from . import flows