diff --git a/CMakeLists.txt b/CMakeLists.txt
index 20f2277d8..22cf39314 100644
--- a/CMakeLists.txt
+++ b/CMakeLists.txt
@@ -20,7 +20,7 @@ option(MLX_ENABLE_X64_MAC "Enable building for x64 macOS" OFF)
 option(BUILD_SHARED_LIBS "Build mlx as a shared library" OFF)
 
 if(NOT MLX_VERSION)
-  set(MLX_VERSION 0.8.1)
+  set(MLX_VERSION 0.9.0)
 endif()
 
 # --------------------- Processor tests -------------------------
diff --git a/setup.py b/setup.py
index eced7e8df..3911a15a8 100644
--- a/setup.py
+++ b/setup.py
@@ -163,7 +163,7 @@ def run(self) -> None:
 
     setup(
         name="mlx",
-        version=get_version("0.8.1"),
+        version=get_version("0.9.0"),
         author="MLX Contributors",
         author_email="mlx@group.apple.com",
         description="A framework for machine learning on Apple silicon.",