diff --git a/CHANGELOG.md b/CHANGELOG.md
index 96081573..c7fffa97 100644
--- a/CHANGELOG.md
+++ b/CHANGELOG.md
@@ -5,6 +5,12 @@ All notable changes to this project will be documented in this file.
 The format is based on [Keep a Changelog](https://keepachangelog.com/en/1.0.0/),
 and this project adheres to [Semantic Versioning](https://semver.org/spec/v2.0.0.html).
 
+## [v0.5.2](https://github.com/elixir-nx/bumblebee/tree/v0.5.2) (2024-02-24)
+
+### Fixed
+
+* Fixed loading sharded parameters in the safetensors format
+
 ## [v0.5.1](https://github.com/elixir-nx/bumblebee/tree/v0.5.1) (2024-02-23)
 
 ### Fixed
diff --git a/README.md b/README.md
index 509d653d..4a8939c4 100644
--- a/README.md
+++ b/README.md
@@ -26,7 +26,7 @@ First add Bumblebee and EXLA as dependencies in your `mix.exs`. EXLA is an optio
 ```elixir
 def deps do
   [
-    {:bumblebee, "~> 0.5.1"},
+    {:bumblebee, "~> 0.5.2"},
     {:exla, ">= 0.0.0"}
   ]
 end
@@ -47,7 +47,7 @@ In notebooks and scripts, use the following `Mix.install/2` call to both install
 ```elixir
 Mix.install(
   [
-    {:bumblebee, "~> 0.5.1"},
+    {:bumblebee, "~> 0.5.2"},
     {:exla, ">= 0.0.0"}
   ],
   config: [nx: [default_backend: EXLA.Backend]]
diff --git a/mix.exs b/mix.exs
index 04f96b48..faa86611 100644
--- a/mix.exs
+++ b/mix.exs
@@ -1,7 +1,7 @@
 defmodule Bumblebee.MixProject do
   use Mix.Project
 
-  @version "0.5.1"
+  @version "0.5.2"
   @description "Pre-trained and transformer Neural Network models in Axon"
 
   def project do