diff --git a/Cargo.lock b/Cargo.lock
index 7214dfec7..803c70aa3 100644
--- a/Cargo.lock
+++ b/Cargo.lock
@@ -304,7 +304,7 @@ dependencies = [
 
 [[package]]
 name = "backend-comparison"
-version = "0.15.0"
+version = "0.16.0"
 dependencies = [
  "arboard",
  "burn",
@@ -490,7 +490,7 @@ checksum = "79296716171880943b8470b5f8d03aa55eb2e645a4874bdbb28adb49162e012c"
 
 [[package]]
 name = "burn"
-version = "0.15.0"
+version = "0.16.0"
 dependencies = [
  "burn-core",
  "burn-train",
@@ -498,7 +498,7 @@ dependencies = [
 
 [[package]]
 name = "burn-autodiff"
-version = "0.15.0"
+version = "0.16.0"
 dependencies = [
  "burn-common",
  "burn-tensor",
@@ -510,7 +510,7 @@ dependencies = [
 
 [[package]]
 name = "burn-candle"
-version = "0.15.0"
+version = "0.16.0"
 dependencies = [
  "burn-autodiff",
  "burn-tch",
@@ -522,7 +522,7 @@ dependencies = [
 
 [[package]]
 name = "burn-common"
-version = "0.15.0"
+version = "0.16.0"
 dependencies = [
  "cubecl-common",
  "dashmap",
@@ -536,7 +536,7 @@ dependencies = [
 
 [[package]]
 name = "burn-core"
-version = "0.15.0"
+version = "0.16.0"
 dependencies = [
  "ahash",
  "bincode",
@@ -572,7 +572,7 @@ dependencies = [
 
 [[package]]
 name = "burn-cuda"
-version = "0.15.0"
+version = "0.16.0"
 dependencies = [
  "burn-fusion",
  "burn-jit",
@@ -586,7 +586,7 @@ dependencies = [
 
 [[package]]
 name = "burn-dataset"
-version = "0.15.0"
+version = "0.16.0"
 dependencies = [
  "burn-common",
  "csv",
@@ -618,7 +618,7 @@ dependencies = [
 
 [[package]]
 name = "burn-derive"
-version = "0.15.0"
+version = "0.16.0"
 dependencies = [
  "derive-new",
  "proc-macro2",
@@ -628,7 +628,7 @@ dependencies = [
 
 [[package]]
 name = "burn-fusion"
-version = "0.15.0"
+version = "0.16.0"
 dependencies = [
  "burn-common",
  "burn-tensor",
@@ -642,7 +642,7 @@ dependencies = [
 
 [[package]]
 name = "burn-hip"
-version = "0.15.0"
+version = "0.16.0"
 dependencies = [
  "burn-fusion",
  "burn-jit",
@@ -656,7 +656,7 @@ dependencies = [
 
 [[package]]
 name = "burn-import"
-version = "0.15.0"
+version = "0.16.0"
 dependencies = [
  "burn",
  "candle-core",
@@ -681,7 +681,7 @@ dependencies = [
 
 [[package]]
 name = "burn-jit"
-version = "0.15.0"
+version = "0.16.0"
 dependencies = [
  "burn-autodiff",
  "burn-common",
@@ -706,7 +706,7 @@ dependencies = [
 
 [[package]]
 name = "burn-ndarray"
-version = "0.15.0"
+version = "0.16.0"
 dependencies = [
  "atomic_float",
  "blas-src",
@@ -726,7 +726,7 @@ dependencies = [
 
 [[package]]
 name = "burn-no-std-tests"
-version = "0.15.0"
+version = "0.16.0"
 dependencies = [
  "burn",
  "burn-ndarray",
@@ -735,7 +735,7 @@ dependencies = [
 
 [[package]]
 name = "burn-router"
-version = "0.15.0"
+version = "0.16.0"
 dependencies = [
  "burn-autodiff",
  "burn-ndarray",
@@ -747,7 +747,7 @@ dependencies = [
 
 [[package]]
 name = "burn-tch"
-version = "0.15.0"
+version = "0.16.0"
 dependencies = [
  "burn-autodiff",
  "burn-tensor",
@@ -760,7 +760,7 @@ dependencies = [
 
 [[package]]
 name = "burn-tensor"
-version = "0.15.0"
+version = "0.16.0"
 dependencies = [
  "burn-common",
  "burn-tensor-testgen",
@@ -780,7 +780,7 @@ dependencies = [
 
 [[package]]
 name = "burn-tensor-testgen"
-version = "0.15.0"
+version = "0.16.0"
 dependencies = [
  "proc-macro2",
  "quote",
@@ -788,7 +788,7 @@ dependencies = [
 
 [[package]]
 name = "burn-train"
-version = "0.15.0"
+version = "0.16.0"
 dependencies = [
  "burn-core",
  "burn-ndarray",
@@ -806,7 +806,7 @@ dependencies = [
 
 [[package]]
 name = "burn-wgpu"
-version = "0.15.0"
+version = "0.16.0"
 dependencies = [
  "burn-fusion",
  "burn-jit",
@@ -1687,7 +1687,7 @@ dependencies = [
 
 [[package]]
 name = "custom-csv-dataset"
-version = "0.15.0"
+version = "0.16.0"
 dependencies = [
  "burn",
  "csv",
@@ -1697,7 +1697,7 @@ dependencies = [
 
 [[package]]
 name = "custom-cubecl-kernel"
-version = "0.15.0"
+version = "0.16.0"
 dependencies = [
  "burn",
  "burn-jit",
@@ -1710,7 +1710,7 @@ dependencies = [
 
 [[package]]
 name = "custom-image-dataset"
-version = "0.15.0"
+version = "0.16.0"
 dependencies = [
  "burn",
  "flate2",
@@ -1719,7 +1719,7 @@ dependencies = [
 
 [[package]]
 name = "custom-renderer"
-version = "0.15.0"
+version = "0.16.0"
 dependencies = [
  "burn",
  "bytemuck",
@@ -1731,7 +1731,7 @@ dependencies = [
 
 [[package]]
 name = "custom-training-loop"
-version = "0.15.0"
+version = "0.16.0"
 dependencies = [
  "burn",
  "bytemuck",
@@ -1743,7 +1743,7 @@ dependencies = [
 
 [[package]]
 name = "custom-wgpu-kernel"
-version = "0.15.0"
+version = "0.16.0"
 dependencies = [
  "burn",
  "bytemuck",
@@ -2793,7 +2793,7 @@ dependencies = [
 
 [[package]]
 name = "guide"
-version = "0.15.0"
+version = "0.16.0"
 dependencies = [
  "burn",
  "log",
@@ -3259,7 +3259,7 @@ dependencies = [
 
 [[package]]
 name = "image-classification-web"
-version = "0.15.0"
+version = "0.16.0"
 dependencies = [
  "burn",
  "burn-candle",
@@ -3780,7 +3780,7 @@ dependencies = [
 
 [[package]]
 name = "mnist"
-version = "0.15.0"
+version = "0.16.0"
 dependencies = [
  "burn",
  "log",
@@ -3789,7 +3789,7 @@ dependencies = [
 
 [[package]]
 name = "mnist-inference-web"
-version = "0.15.0"
+version = "0.16.0"
 dependencies = [
  "burn",
  "console_error_panic_hook",
@@ -3802,7 +3802,7 @@ dependencies = [
 
 [[package]]
 name = "model"
-version = "0.4.0"
+version = "0.5.0"
 dependencies = [
  "burn",
  "burn-import",
@@ -3874,7 +3874,7 @@ dependencies = [
 
 [[package]]
 name = "named-tensor"
-version = "0.15.0"
+version = "0.16.0"
 dependencies = [
  "burn",
  "serde",
@@ -4320,7 +4320,7 @@ dependencies = [
 
 [[package]]
 name = "onnx-inference"
-version = "0.15.0"
+version = "0.16.0"
 dependencies = [
  "burn",
  "burn-import",
@@ -4329,7 +4329,7 @@ dependencies = [
 
 [[package]]
 name = "onnx-ir"
-version = "0.15.0"
+version = "0.16.0"
 dependencies = [
  "bytemuck",
  "half",
@@ -4346,7 +4346,7 @@ dependencies = [
 
 [[package]]
 name = "onnx-tests"
-version = "0.15.0"
+version = "0.16.0"
 dependencies = [
  "burn",
  "burn-import",
@@ -5244,7 +5244,7 @@ dependencies = [
 
 [[package]]
 name = "pytorch-import"
-version = "0.15.0"
+version = "0.16.0"
 dependencies = [
  "burn",
  "burn-import",
@@ -5253,7 +5253,7 @@ dependencies = [
 
 [[package]]
 name = "pytorch-tests"
-version = "0.15.0"
+version = "0.16.0"
 dependencies = [
  "burn",
  "burn-autodiff",
@@ -6211,7 +6211,7 @@ checksum = "e3a9fe34e3e7a50316060351f37187a3f546bce95496156754b601a5fa71b76e"
 
 [[package]]
 name = "simple-regression"
-version = "0.15.0"
+version = "0.16.0"
 dependencies = [
  "burn",
  "log",
@@ -6617,7 +6617,7 @@ dependencies = [
 
 [[package]]
 name = "text-classification"
-version = "0.15.0"
+version = "0.16.0"
 dependencies = [
  "burn",
  "derive-new",
@@ -6627,7 +6627,7 @@ dependencies = [
 
 [[package]]
 name = "text-generation"
-version = "0.15.0"
+version = "0.16.0"
 dependencies = [
  "burn",
  "derive-new",
@@ -7770,7 +7770,7 @@ checksum = "af4e2e2f7cba5a093896c1e150fbfe177d1883e7448200efb81d40b9d339ef26"
 
 [[package]]
 name = "xtask"
-version = "1.0.0"
+version = "1.1.0"
 dependencies = [
  "log",
  "rstest",
diff --git a/Cargo.toml b/Cargo.toml
index 9b9147dd3..af68c1b44 100644
--- a/Cargo.toml
+++ b/Cargo.toml
@@ -24,7 +24,7 @@ exclude = [
 edition = "2021"
 license = "MIT OR Apache-2.0"
 readme = "README.md"
-version = "0.15.0"
+version = "0.16.0"
 
 [workspace.dependencies]
 atomic_float = "1"
diff --git a/backend-comparison/Cargo.toml b/backend-comparison/Cargo.toml
index a64b9675d..9e199ba8c 100644
--- a/backend-comparison/Cargo.toml
+++ b/backend-comparison/Cargo.toml
@@ -33,8 +33,8 @@ wgpu-spirv-fusion = ["wgpu-spirv", "burn/fusion"]
 [dependencies]
 arboard = { workspace = true }
 burn = { path = "../crates/burn", default-features = false }
-burn-common = { path = "../crates/burn-common", version = "0.15.0" }
-burn-wgpu = { path = "../crates/burn-wgpu", default-features = false, version = "0.15.0", optional = true }
+burn-common = { path = "../crates/burn-common", version = "0.16.0" }
+burn-wgpu = { path = "../crates/burn-wgpu", default-features = false, version = "0.16.0", optional = true }
 clap = { workspace = true }
 colored = { workspace = true }
 cubecl = { workspace = true, features = ["wgpu"], default-features = true }
diff --git a/burn-book/src/advanced/no-std.md b/burn-book/src/advanced/no-std.md
index 14b8a1b87..7689d2535 100644
--- a/burn-book/src/advanced/no-std.md
+++ b/burn-book/src/advanced/no-std.md
@@ -23,7 +23,7 @@ Some other dependencies have to be added
 ```toml
 [dependencies]
 embedded-alloc = "0.5.1" # Only if there is no default allocator for your chip
-burn = { version = "0.15", default-features = false, features = ["ndarray"] } # Backend must be ndarray
+burn = { version = "0.16", default-features = false, features = ["ndarray"] } # Backend must be ndarray
 
 [build-dependencies]
 burn-import = { version = "0.14" } # Used to auto generate the rust code to import the model
diff --git a/burn-book/src/basic-workflow/README.md b/burn-book/src/basic-workflow/README.md
index 78cb46d40..5b32591a5 100644
--- a/burn-book/src/basic-workflow/README.md
+++ b/burn-book/src/basic-workflow/README.md
@@ -14,7 +14,7 @@ automatically add the missing imports as you add the code snippets to your code.
 Be sure to checkout the git branch corresponding to the version of Burn you are using to follow
 this guide.
 
-The current version of Burn is `0.15` and the corresponding branch to checkout is `main`.
+The current version of Burn is `0.16` and the corresponding branch to checkout is `main`.
 </div>
 
 The code for this demo can be executed from Burn's base directory using the command:
diff --git a/burn-book/src/basic-workflow/model.md b/burn-book/src/basic-workflow/model.md
index e8471fa12..adce46b29 100644
--- a/burn-book/src/basic-workflow/model.md
+++ b/burn-book/src/basic-workflow/model.md
@@ -20,7 +20,7 @@ version = "0.1.0"
 edition = "2021"
 
 [dependencies]
-burn = { version = "~0.15", features = ["train", "wgpu", "vision"] }
+burn = { version = "~0.16", features = ["train", "wgpu", "vision"] }
 ```
 
 Our goal will be to create a basic convolutional neural network used for image classification. We
diff --git a/burn-book/src/import/onnx-model.md b/burn-book/src/import/onnx-model.md
index e9deebaa0..05b9d5de8 100644
--- a/burn-book/src/import/onnx-model.md
+++ b/burn-book/src/import/onnx-model.md
@@ -74,7 +74,7 @@ First, add the `burn-import` crate to your `Cargo.toml`:
 
 ```toml
 [build-dependencies]
-burn-import = "~0.15"
+burn-import = "~0.16"
 ```
 
 Then, in your `build.rs` file:
diff --git a/crates/burn-autodiff/Cargo.toml b/crates/burn-autodiff/Cargo.toml
index 49c0b0906..2144d4688 100644
--- a/crates/burn-autodiff/Cargo.toml
+++ b/crates/burn-autodiff/Cargo.toml
@@ -18,16 +18,16 @@ std = []
 async = [] # Require std
 
 [dependencies]
-burn-common = { path = "../burn-common", version = "0.15.0" }
-burn-tensor = { path = "../burn-tensor", version = "0.15.0", default-features = false }
-burn-tensor-testgen = { path = "../burn-tensor-testgen", version = "0.15.0", optional = true }
+burn-common = { path = "../burn-common", version = "0.16.0" }
+burn-tensor = { path = "../burn-tensor", version = "0.16.0", default-features = false }
+burn-tensor-testgen = { path = "../burn-tensor-testgen", version = "0.16.0", optional = true }
 
 derive-new = { workspace = true }
 spin = { workspace = true }
 log = { workspace = true }
 
 [dev-dependencies]
-burn-tensor = { path = "../burn-tensor", version = "0.15.0", default-features = false, features = [
+burn-tensor = { path = "../burn-tensor", version = "0.16.0", default-features = false, features = [
   "export_tests",
 ] }
 
diff --git a/crates/burn-candle/Cargo.toml b/crates/burn-candle/Cargo.toml
index 0abad2905..62af31d5f 100644
--- a/crates/burn-candle/Cargo.toml
+++ b/crates/burn-candle/Cargo.toml
@@ -21,17 +21,17 @@ accelerate = ["candle-core/accelerate"]
 
 [dependencies]
 derive-new = { workspace = true }
-burn-tensor = { path = "../burn-tensor", version = "0.15.0", default-features = false }
+burn-tensor = { path = "../burn-tensor", version = "0.16.0", default-features = false }
 half = { workspace = true }
 candle-core = { workspace = true }
 
 [dev-dependencies]
-burn-autodiff = { path = "../burn-autodiff", version = "0.15.0", default-features = false, features = [
+burn-autodiff = { path = "../burn-autodiff", version = "0.16.0", default-features = false, features = [
     "export_tests",
 ] }
-burn-tch = { path = "../burn-tch", version = "0.15.0", default-features = false, features = [
+burn-tch = { path = "../burn-tch", version = "0.16.0", default-features = false, features = [
 ] }
-burn-tensor = { path = "../burn-tensor", version = "0.15.0", default-features = false, features = [
+burn-tensor = { path = "../burn-tensor", version = "0.16.0", default-features = false, features = [
     "export_tests",
 ] }
 
diff --git a/crates/burn-core/Cargo.toml b/crates/burn-core/Cargo.toml
index b4d03c4ce..88d52370a 100644
--- a/crates/burn-core/Cargo.toml
+++ b/crates/burn-core/Cargo.toml
@@ -118,19 +118,19 @@ test-wgpu-spirv = [
 
 # ** Please make sure all dependencies support no_std when std is disabled **
 
-burn-common = { path = "../burn-common", version = "0.15.0", default-features = false }
-burn-dataset = { path = "../burn-dataset", version = "0.15.0", optional = true, default-features = false }
-burn-derive = { path = "../burn-derive", version = "0.15.0" }
-burn-tensor = { path = "../burn-tensor", version = "0.15.0", default-features = false }
+burn-common = { path = "../burn-common", version = "0.16.0", default-features = false }
+burn-dataset = { path = "../burn-dataset", version = "0.16.0", optional = true, default-features = false }
+burn-derive = { path = "../burn-derive", version = "0.16.0" }
+burn-tensor = { path = "../burn-tensor", version = "0.16.0", default-features = false }
 
 # Backends
-burn-autodiff = { path = "../burn-autodiff", version = "0.15.0", optional = true }
-burn-candle = { path = "../burn-candle", version = "0.15.0", optional = true }
-burn-cuda = { path = "../burn-cuda", version = "0.15.0", optional = true, default-features = false }
-burn-hip = { path = "../burn-hip", version = "0.15.0", optional = true, default-features = false }
-burn-ndarray = { path = "../burn-ndarray", version = "0.15.0", optional = true, default-features = false }
-burn-tch = { path = "../burn-tch", version = "0.15.0", optional = true }
-burn-wgpu = { path = "../burn-wgpu", version = "0.15.0", optional = true, default-features = false }
+burn-autodiff = { path = "../burn-autodiff", version = "0.16.0", optional = true }
+burn-candle = { path = "../burn-candle", version = "0.16.0", optional = true }
+burn-cuda = { path = "../burn-cuda", version = "0.16.0", optional = true, default-features = false }
+burn-hip = { path = "../burn-hip", version = "0.16.0", optional = true, default-features = false }
+burn-ndarray = { path = "../burn-ndarray", version = "0.16.0", optional = true, default-features = false }
+burn-tch = { path = "../burn-tch", version = "0.16.0", optional = true }
+burn-wgpu = { path = "../burn-wgpu", version = "0.16.0", optional = true, default-features = false }
 
 data-encoding = { workspace = true }
 uuid = { workspace = true }
@@ -160,13 +160,13 @@ thiserror = { workspace = true, optional = true }
 portable-atomic-util = { workspace = true }
 
 [dev-dependencies]
-burn-dataset = { path = "../burn-dataset", version = "0.15.0", features = [
+burn-dataset = { path = "../burn-dataset", version = "0.16.0", features = [
     "fake",
 ] }
 tempfile = { workspace = true }
 
-burn-autodiff = { path = "../burn-autodiff", version = "0.15.0" }
-burn-ndarray = { path = "../burn-ndarray", version = "0.15.0", default-features = false }
+burn-autodiff = { path = "../burn-autodiff", version = "0.16.0" }
+burn-ndarray = { path = "../burn-ndarray", version = "0.16.0", default-features = false }
 
 [package.metadata.docs.rs]
 features = ["doc"]
diff --git a/crates/burn-cuda/Cargo.toml b/crates/burn-cuda/Cargo.toml
index 1f837c78e..4a5456e1a 100644
--- a/crates/burn-cuda/Cargo.toml
+++ b/crates/burn-cuda/Cargo.toml
@@ -20,9 +20,9 @@ std = ["burn-jit/std", "cubecl/std"]
 
 [dependencies]
 cubecl = { workspace = true, features = ["cuda"] }
-burn-jit = { path = "../burn-jit", version = "0.15.0", default-features = false }
-burn-tensor = { path = "../burn-tensor", version = "0.15.0", features = ["cubecl-cuda"] }
-burn-fusion = { path = "../burn-fusion", version = "0.15.0", optional = true }
+burn-jit = { path = "../burn-jit", version = "0.16.0", default-features = false }
+burn-tensor = { path = "../burn-tensor", version = "0.16.0", features = ["cubecl-cuda"] }
+burn-fusion = { path = "../burn-fusion", version = "0.16.0", optional = true }
 
 half = { workspace = true }
 bytemuck = { workspace = true }
@@ -31,7 +31,7 @@ log = { workspace = true }
 derive-new = { workspace = true }
 
 [dev-dependencies]
-burn-jit = { path = "../burn-jit", version = "0.15.0", default-features = false, features = [
+burn-jit = { path = "../burn-jit", version = "0.16.0", default-features = false, features = [
   "export_tests",
 ] }
 
diff --git a/crates/burn-dataset/Cargo.toml b/crates/burn-dataset/Cargo.toml
index c506eaa99..023776597 100644
--- a/crates/burn-dataset/Cargo.toml
+++ b/crates/burn-dataset/Cargo.toml
@@ -30,7 +30,7 @@ __sqlite-shared = [
 dataframe = ["dep:polars"]
 
 [dependencies]
-burn-common = { path = "../burn-common", version = "0.15.0", optional = true, features = [
+burn-common = { path = "../burn-common", version = "0.16.0", optional = true, features = [
     "network",
 ] }
 csv = { workspace = true }
diff --git a/crates/burn-fusion/Cargo.toml b/crates/burn-fusion/Cargo.toml
index 27b312f49..eb4296097 100644
--- a/crates/burn-fusion/Cargo.toml
+++ b/crates/burn-fusion/Cargo.toml
@@ -17,8 +17,8 @@ std = ["serde/std"]
 doc = ["default"]
 
 [dependencies]
-burn-tensor = { path = "../burn-tensor", version = "0.15.0" }
-burn-common = { path = "../burn-common", version = "0.15.0" }
+burn-tensor = { path = "../burn-tensor", version = "0.16.0" }
+burn-common = { path = "../burn-common", version = "0.16.0" }
 hashbrown = { workspace = true }
 derive-new = {workspace = true }
 spin = { workspace = true }
diff --git a/crates/burn-hip/Cargo.toml b/crates/burn-hip/Cargo.toml
index ced4e13b8..65ae4a0b0 100644
--- a/crates/burn-hip/Cargo.toml
+++ b/crates/burn-hip/Cargo.toml
@@ -20,9 +20,9 @@ std = ["burn-jit/std", "cubecl/std"]
 
 [dependencies]
 cubecl = { workspace = true, features = ["hip"] }
-burn-jit = { path = "../burn-jit", version = "0.15.0", default-features = false }
-burn-tensor = { path = "../burn-tensor", version = "0.15.0", features = ["cubecl-hip"] }
-burn-fusion = { path = "../burn-fusion", version = "0.15.0", optional = true }
+burn-jit = { path = "../burn-jit", version = "0.16.0", default-features = false }
+burn-tensor = { path = "../burn-tensor", version = "0.16.0", features = ["cubecl-hip"] }
+burn-fusion = { path = "../burn-fusion", version = "0.16.0", optional = true }
 
 half = { workspace = true }
 bytemuck = { workspace = true }
@@ -31,7 +31,7 @@ log = { workspace = true }
 derive-new = { workspace = true }
 
 [dev-dependencies]
-burn-jit = { path = "../burn-jit", version = "0.15.0", default-features = false, features = [
+burn-jit = { path = "../burn-jit", version = "0.16.0", default-features = false, features = [
   "export_tests",
 ] }
 
diff --git a/crates/burn-import/Cargo.toml b/crates/burn-import/Cargo.toml
index 0f7078b19..a75985e71 100644
--- a/crates/burn-import/Cargo.toml
+++ b/crates/burn-import/Cargo.toml
@@ -20,8 +20,8 @@ onnx = []
 pytorch = ["burn/record-item-custom-serde", "thiserror", "zip"]
 
 [dependencies]
-burn = { path = "../burn", version = "0.15.0", features = ["ndarray"] }
-onnx-ir = { path = "../onnx-ir", version = "0.15.0" }
+burn = { path = "../burn", version = "0.16.0", features = ["ndarray"] }
+onnx-ir = { path = "../onnx-ir", version = "0.16.0" }
 candle-core = { workspace = true }
 derive-new = { workspace = true }
 half = { workspace = true }
diff --git a/crates/burn-jit/Cargo.toml b/crates/burn-jit/Cargo.toml
index c6463baa3..716ba6a82 100644
--- a/crates/burn-jit/Cargo.toml
+++ b/crates/burn-jit/Cargo.toml
@@ -28,9 +28,9 @@ std = ["cubecl/std"]
 template = []
 
 [dependencies]
-burn-common = { path = "../burn-common", version = "0.15.0" }
-burn-fusion = { path = "../burn-fusion", version = "0.15.0", optional = true }
-burn-tensor = { path = "../burn-tensor", version = "0.15.0", features = [
+burn-common = { path = "../burn-common", version = "0.16.0" }
+burn-fusion = { path = "../burn-fusion", version = "0.16.0", optional = true }
+burn-tensor = { path = "../burn-tensor", version = "0.16.0", features = [
   "cubecl", "repr",
 ] }
 cubecl = { workspace = true, features = ["linalg"] }
@@ -50,12 +50,12 @@ futures-lite = { workspace = true, features = ["std"] }
 serde = { workspace = true }
 text_placeholder = { workspace = true, features = ["struct_context"] }
 
-burn-tensor-testgen = { path = "../burn-tensor-testgen", version = "0.15.0", optional = true }
+burn-tensor-testgen = { path = "../burn-tensor-testgen", version = "0.16.0", optional = true }
 hashbrown = { workspace = true }
 
 # When exporting tests
-burn-autodiff = { path = "../burn-autodiff", version = "0.15.0", default-features = false, optional = true }
-burn-ndarray = { path = "../burn-ndarray", version = "0.15.0", optional = true }
+burn-autodiff = { path = "../burn-autodiff", version = "0.16.0", default-features = false, optional = true }
+burn-ndarray = { path = "../burn-ndarray", version = "0.16.0", optional = true }
 serial_test = { workspace = true, optional = true }
 
 [package.metadata.docs.rs]
diff --git a/crates/burn-ndarray/Cargo.toml b/crates/burn-ndarray/Cargo.toml
index d41c37ed5..89253cd7e 100644
--- a/crates/burn-ndarray/Cargo.toml
+++ b/crates/burn-ndarray/Cargo.toml
@@ -43,9 +43,9 @@ blas-openblas-system = [
 
 # ** Please make sure all dependencies support no_std when std is disabled **
 
-burn-autodiff = { path = "../burn-autodiff", version = "0.15.0", optional = true }
-burn-common = { path = "../burn-common", version = "0.15.0", default-features = false }
-burn-tensor = { path = "../burn-tensor", version = "0.15.0", default-features = false, features = ["repr"] }
+burn-autodiff = { path = "../burn-autodiff", version = "0.16.0", optional = true }
+burn-common = { path = "../burn-common", version = "0.16.0", default-features = false }
+burn-tensor = { path = "../burn-tensor", version = "0.16.0", default-features = false, features = ["repr"] }
 
 atomic_float = { workspace = true }
 blas-src = { workspace = true, default-features = false, optional = true } # no-std compatible
@@ -62,10 +62,10 @@ spin = { workspace = true }                                                # usi
 portable-atomic-util = { workspace = true }
 
 [dev-dependencies]
-burn-autodiff = { path = "../burn-autodiff", version = "0.15.0", default-features = false, features = [
+burn-autodiff = { path = "../burn-autodiff", version = "0.16.0", default-features = false, features = [
     "export_tests",
 ] }
-burn-tensor = { path = "../burn-tensor", version = "0.15.0", default-features = false, features = [
+burn-tensor = { path = "../burn-tensor", version = "0.16.0", default-features = false, features = [
     "export_tests",
 ] }
 
diff --git a/crates/burn-no-std-tests/Cargo.toml b/crates/burn-no-std-tests/Cargo.toml
index 4fe2cfe75..e15ce56d1 100644
--- a/crates/burn-no-std-tests/Cargo.toml
+++ b/crates/burn-no-std-tests/Cargo.toml
@@ -14,7 +14,7 @@ version.workspace = true
 
 # ** Please make sure all dependencies support no_std **
 
-burn = { path = "../burn", version = "0.15.0", default-features = false }
-burn-ndarray = { path = "../burn-ndarray", version = "0.15.0", default-features = false }
+burn = { path = "../burn", version = "0.16.0", default-features = false }
+burn-ndarray = { path = "../burn-ndarray", version = "0.16.0", default-features = false }
 
 serde = { workspace = true }
diff --git a/crates/burn-router/Cargo.toml b/crates/burn-router/Cargo.toml
index aa93faf1f..efe2c3da0 100644
--- a/crates/burn-router/Cargo.toml
+++ b/crates/burn-router/Cargo.toml
@@ -17,21 +17,21 @@ std = []
 doc = ["default"]
 
 [dependencies]
-burn-tensor = { path = "../burn-tensor", version = "0.15.0", default-features = false, features = ["repr"]}
+burn-tensor = { path = "../burn-tensor", version = "0.16.0", default-features = false, features = ["repr"]}
 hashbrown = { workspace = true }
 spin = { workspace = true }
 
 
 [dev-dependencies]
-burn-autodiff = { path = "../burn-autodiff", version = "0.15.0", default-features = false, features = [
+burn-autodiff = { path = "../burn-autodiff", version = "0.16.0", default-features = false, features = [
   "export_tests",
 ] }
-burn-tensor = { path = "../burn-tensor", version = "0.15.0", default-features = false, features = [
+burn-tensor = { path = "../burn-tensor", version = "0.16.0", default-features = false, features = [
   "export_tests",
 ] }
 
-burn-ndarray = { path = "../burn-ndarray", version = "0.15.0" }
-burn-wgpu = { path = "../burn-wgpu", version = "0.15.0" }
+burn-ndarray = { path = "../burn-ndarray", version = "0.16.0" }
+burn-wgpu = { path = "../burn-wgpu", version = "0.16.0" }
 
 
 [package.metadata.docs.rs]
diff --git a/crates/burn-tch/Cargo.toml b/crates/burn-tch/Cargo.toml
index 239b969b4..44702c21b 100644
--- a/crates/burn-tch/Cargo.toml
+++ b/crates/burn-tch/Cargo.toml
@@ -16,7 +16,7 @@ default = []
 doc = ["tch/doc-only"]
 
 [dependencies]
-burn-tensor = { path = "../burn-tensor", version = "0.15.0" }
+burn-tensor = { path = "../burn-tensor", version = "0.16.0" }
 
 half = { workspace = true, features = ["std"] }
 libc = { workspace = true }
@@ -25,10 +25,10 @@ tch = { workspace = true, features = ["download-libtorch"] }
 log = { workspace = true }
 
 [dev-dependencies]
-burn-autodiff = { path = "../burn-autodiff", version = "0.15.0", default-features = false, features = [
+burn-autodiff = { path = "../burn-autodiff", version = "0.16.0", default-features = false, features = [
   "export_tests",
 ] }
-burn-tensor = { path = "../burn-tensor", version = "0.15.0", default-features = false, features = [
+burn-tensor = { path = "../burn-tensor", version = "0.16.0", default-features = false, features = [
   "export_tests",
 ] }
 
diff --git a/crates/burn-tensor/Cargo.toml b/crates/burn-tensor/Cargo.toml
index b89b9a5d9..cc7e14e9b 100644
--- a/crates/burn-tensor/Cargo.toml
+++ b/crates/burn-tensor/Cargo.toml
@@ -31,8 +31,8 @@ cubecl-cuda = ["cubecl", "cubecl/cuda"]
 cubecl-hip = ["cubecl", "cubecl/hip"]
 
 [dependencies]
-burn-common = { path = "../burn-common", version = "0.15.0", default-features = false }
-burn-tensor-testgen = { path = "../burn-tensor-testgen", version = "0.15.0", optional = true }
+burn-common = { path = "../burn-common", version = "0.16.0", default-features = false }
+burn-tensor-testgen = { path = "../burn-tensor-testgen", version = "0.16.0", optional = true }
 cubecl = { workspace = true, optional = true }
 
 derive-new = { workspace = true }
diff --git a/crates/burn-train/Cargo.toml b/crates/burn-train/Cargo.toml
index c5b47e441..88ef13eb7 100644
--- a/crates/burn-train/Cargo.toml
+++ b/crates/burn-train/Cargo.toml
@@ -18,7 +18,7 @@ metrics = ["nvml-wrapper", "sysinfo", "systemstat"]
 tui = ["ratatui"]
 
 [dependencies]
-burn-core = { path = "../burn-core", version = "0.15.0", features = [
+burn-core = { path = "../burn-core", version = "0.16.0", features = [
     "dataset",
     "std",
 ], default-features = false }
@@ -41,7 +41,7 @@ derive-new = { workspace = true }
 serde = { workspace = true, features = ["std", "derive"] }
 
 [dev-dependencies]
-burn-ndarray = { path = "../burn-ndarray", version = "0.15.0" }
+burn-ndarray = { path = "../burn-ndarray", version = "0.16.0" }
 
 [package.metadata.docs.rs]
 features = ["doc"]
diff --git a/crates/burn-wgpu/Cargo.toml b/crates/burn-wgpu/Cargo.toml
index f65ea9574..1e94d1f97 100644
--- a/crates/burn-wgpu/Cargo.toml
+++ b/crates/burn-wgpu/Cargo.toml
@@ -24,15 +24,15 @@ template = ["burn-jit/template", "cubecl/template"]
 [dependencies]
 cubecl = { workspace = true, features = ["wgpu"] }
 
-burn-fusion = { path = "../burn-fusion", version = "0.15.0", optional = true }
-burn-jit = { path = "../burn-jit", version = "0.15.0", default-features = false }
-burn-tensor = { path = "../burn-tensor", version = "0.15.0", features = [
+burn-fusion = { path = "../burn-fusion", version = "0.16.0", optional = true }
+burn-jit = { path = "../burn-jit", version = "0.16.0", default-features = false }
+burn-tensor = { path = "../burn-tensor", version = "0.16.0", features = [
   "cubecl-wgpu",
 ] }
 
 
 [dev-dependencies]
-burn-jit = { path = "../burn-jit", version = "0.15.0", default-features = false, features = [
+burn-jit = { path = "../burn-jit", version = "0.16.0", default-features = false, features = [
   "export_tests",
 ] }
 
diff --git a/crates/burn/Cargo.toml b/crates/burn/Cargo.toml
index a52d63ce9..34038a946 100644
--- a/crates/burn/Cargo.toml
+++ b/crates/burn/Cargo.toml
@@ -70,5 +70,5 @@ record-item-custom-serde = ["burn-core/record-item-custom-serde"]
 
 # ** Please make sure all dependencies support no_std when std is disabled **
 
-burn-core = { path = "../burn-core", version = "0.15.0", default-features = false }
-burn-train = { path = "../burn-train", version = "0.15.0", optional = true, default-features = false }
+burn-core = { path = "../burn-core", version = "0.16.0", default-features = false }
+burn-train = { path = "../burn-train", version = "0.16.0", optional = true, default-features = false }
diff --git a/examples/image-classification-web/Cargo.toml b/examples/image-classification-web/Cargo.toml
index a71920d70..5f036532a 100644
--- a/examples/image-classification-web/Cargo.toml
+++ b/examples/image-classification-web/Cargo.toml
@@ -14,11 +14,11 @@ default = []
 half_precision = []
 
 [dependencies]
-burn = { path = "../../crates/burn", version = "0.15.0", default-features = false, features = [
+burn = { path = "../../crates/burn", version = "0.16.0", default-features = false, features = [
     "ndarray", "wgpu",
 ] }
 cubecl-runtime = { version = "0.3.0", features = ["channel-mpsc"] } # missing feature flags
-burn-candle = { path = "../../crates/burn-candle", version = "0.15.0", default-features = false }
+burn-candle = { path = "../../crates/burn-candle", version = "0.16.0", default-features = false }
 
 log = { workspace = true }
 serde = { workspace = true }
diff --git a/examples/pytorch-import/Cargo.toml b/examples/pytorch-import/Cargo.toml
index 72b662b4f..a7b330568 100644
--- a/examples/pytorch-import/Cargo.toml
+++ b/examples/pytorch-import/Cargo.toml
@@ -4,7 +4,7 @@ edition = "2021"
 license = "MIT OR Apache-2.0"
 name = "pytorch-import"
 publish = false
-version = "0.15.0"
+version = "0.16.0"
 
 [dependencies]
 burn = { path = "../../crates/burn", features = [
diff --git a/examples/pytorch-import/model/Cargo.toml b/examples/pytorch-import/model/Cargo.toml
index 11867e70e..894ac7e48 100644
--- a/examples/pytorch-import/model/Cargo.toml
+++ b/examples/pytorch-import/model/Cargo.toml
@@ -1,6 +1,6 @@
 [package]
 name = "model"
-version = "0.4.0"
+version = "0.5.0"
 edition = "2021"
 
 [dependencies]
diff --git a/examples/raspberry-pi-pico/Cargo.lock b/examples/raspberry-pi-pico/Cargo.lock
index d2001f7d4..2cbc8fb72 100644
--- a/examples/raspberry-pi-pico/Cargo.lock
+++ b/examples/raspberry-pi-pico/Cargo.lock
@@ -286,7 +286,7 @@ checksum = "79296716171880943b8470b5f8d03aa55eb2e645a4874bdbb28adb49162e012c"
 
 [[package]]
 name = "burn"
-version = "0.15.0"
+version = "0.16.0"
 dependencies = [
  "burn-core",
  "burn-train",
@@ -294,7 +294,7 @@ dependencies = [
 
 [[package]]
 name = "burn-autodiff"
-version = "0.15.0"
+version = "0.16.0"
 dependencies = [
  "burn-common",
  "burn-tensor",
@@ -305,7 +305,7 @@ dependencies = [
 
 [[package]]
 name = "burn-candle"
-version = "0.15.0"
+version = "0.16.0"
 dependencies = [
  "burn-tensor",
  "candle-core",
@@ -315,7 +315,7 @@ dependencies = [
 
 [[package]]
 name = "burn-common"
-version = "0.15.0"
+version = "0.16.0"
 dependencies = [
  "cubecl-common",
  "data-encoding",
@@ -326,7 +326,7 @@ dependencies = [
 
 [[package]]
 name = "burn-core"
-version = "0.15.0"
+version = "0.16.0"
 dependencies = [
  "bincode",
  "burn-autodiff",
@@ -357,7 +357,7 @@ dependencies = [
 
 [[package]]
 name = "burn-cuda"
-version = "0.15.0"
+version = "0.16.0"
 dependencies = [
  "burn-fusion",
  "burn-jit",
@@ -371,7 +371,7 @@ dependencies = [
 
 [[package]]
 name = "burn-dataset"
-version = "0.15.0"
+version = "0.16.0"
 dependencies = [
  "csv",
  "derive-new",
@@ -395,7 +395,7 @@ dependencies = [
 
 [[package]]
 name = "burn-derive"
-version = "0.15.0"
+version = "0.16.0"
 dependencies = [
  "derive-new",
  "proc-macro2",
@@ -405,7 +405,7 @@ dependencies = [
 
 [[package]]
 name = "burn-fusion"
-version = "0.15.0"
+version = "0.16.0"
 dependencies = [
  "burn-common",
  "burn-tensor",
@@ -418,7 +418,7 @@ dependencies = [
 
 [[package]]
 name = "burn-import"
-version = "0.15.0"
+version = "0.16.0"
 dependencies = [
  "burn",
  "candle-core",
@@ -441,7 +441,7 @@ dependencies = [
 
 [[package]]
 name = "burn-jit"
-version = "0.15.0"
+version = "0.16.0"
 dependencies = [
  "burn-common",
  "burn-fusion",
@@ -461,7 +461,7 @@ dependencies = [
 
 [[package]]
 name = "burn-ndarray"
-version = "0.15.0"
+version = "0.16.0"
 dependencies = [
  "burn-autodiff",
  "burn-common",
@@ -478,7 +478,7 @@ dependencies = [
 
 [[package]]
 name = "burn-tch"
-version = "0.15.0"
+version = "0.16.0"
 dependencies = [
  "burn-tensor",
  "half",
@@ -489,7 +489,7 @@ dependencies = [
 
 [[package]]
 name = "burn-tensor"
-version = "0.15.0"
+version = "0.16.0"
 dependencies = [
  "burn-common",
  "bytemuck",
@@ -507,7 +507,7 @@ dependencies = [
 
 [[package]]
 name = "burn-train"
-version = "0.15.0"
+version = "0.16.0"
 dependencies = [
  "burn-core",
  "crossterm",
@@ -525,7 +525,7 @@ dependencies = [
 
 [[package]]
 name = "burn-wgpu"
-version = "0.15.0"
+version = "0.16.0"
 dependencies = [
  "burn-fusion",
  "burn-jit",
@@ -2959,7 +2959,7 @@ checksum = "3fdb12b2476b595f9358c5161aa467c2438859caa136dec86c26fdd2efe17b92"
 
 [[package]]
 name = "onnx-ir"
-version = "0.15.0"
+version = "0.16.0"
 dependencies = [
  "bytemuck",
  "half",
diff --git a/xtask/Cargo.toml b/xtask/Cargo.toml
index f85a8ae6e..ce796eb7b 100644
--- a/xtask/Cargo.toml
+++ b/xtask/Cargo.toml
@@ -1,6 +1,6 @@
 [package]
 name = "xtask"
-version = "1.0.0"
+version = "1.1.0"
 edition = "2021"
 license = "MIT OR Apache-2.0"