diff --git a/agml/__init__.py b/agml/__init__.py
index 9e33c028..c719984c 100644
--- a/agml/__init__.py
+++ b/agml/__init__.py
@@ -12,7 +12,7 @@
 # See the License for the specific language governing permissions and
 # limitations under the License.
 
-__version__ = '0.1.0'
+__version__ = '0.2.0'
 __all__ = ['data', 'backend', 'viz']
 
 # If AgML is being imported for the first time, then we need to setup