diff --git a/CMakeLists.txt b/CMakeLists.txt
index 2d6bef9d2..b008e10e6 100644
--- a/CMakeLists.txt
+++ b/CMakeLists.txt
@@ -18,7 +18,7 @@ option(MLX_BUILD_METAL "Build metal backend" ON)
 option(BUILD_SHARED_LIBS "Build mlx as a shared library" OFF)
 
 if(NOT MLX_VERSION)
-  set(MLX_VERSION 0.1.0)
+  set(MLX_VERSION 0.2.0)
 endif()
 
 # --------------------- Processor tests -------------------------
diff --git a/setup.py b/setup.py
index 44cccf62d..cce1f8537 100644
--- a/setup.py
+++ b/setup.py
@@ -152,7 +152,7 @@ def run(self) -> None:
 
     setup(
         name="mlx",
-        version=get_version("0.1.0"),
+        version=get_version("0.2.0"),
         author="MLX Contributors",
         author_email="mlx@group.apple.com",
         description="A framework for machine learning on Apple silicon.",