diff --git a/Cargo.lock b/Cargo.lock
index 67db365b9..7214dfec7 100644
--- a/Cargo.lock
+++ b/Cargo.lock
@@ -3264,9 +3264,8 @@ dependencies = [
  "burn",
  "burn-candle",
  "burn-import",
- "burn-wgpu",
  "console_error_panic_hook",
- "cubecl",
+ "cubecl-runtime",
  "js-sys",
  "log",
  "serde",
@@ -3794,6 +3793,7 @@ version = "0.15.0"
 dependencies = [
  "burn",
  "console_error_panic_hook",
+ "cubecl-runtime",
  "js-sys",
  "serde",
  "wasm-bindgen",
diff --git a/examples/image-classification-web/Cargo.toml b/examples/image-classification-web/Cargo.toml
index c000e95f0..a71920d70 100644
--- a/examples/image-classification-web/Cargo.toml
+++ b/examples/image-classification-web/Cargo.toml
@@ -15,12 +15,9 @@ half_precision = []
 
 [dependencies]
 burn = { path = "../../crates/burn", version = "0.15.0", default-features = false, features = [
-    "ndarray",
-] }
-cubecl = { workspace = true, features = ["wgpu"] }
-burn-wgpu = { path = "../../crates/burn-wgpu", version = "0.15.0", default-features = false, features = [
-    "autotune",
+    "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 }
 
 log = { workspace = true }
diff --git a/examples/image-classification-web/src/web.rs b/examples/image-classification-web/src/web.rs
index 84bc75194..c44719006 100644
--- a/examples/image-classification-web/src/web.rs
+++ b/examples/image-classification-web/src/web.rs
@@ -10,9 +10,8 @@ use crate::model::{label::LABELS, normalizer::Normalizer, squeezenet::Model as S
 
 use burn::{backend::NdArray, prelude::*, tensor::activation::softmax};
 
+use burn::backend::wgpu::{init_async, AutoGraphicsApi, Wgpu, WgpuDevice};
 use burn_candle::Candle;
-use burn_wgpu::{Wgpu, WgpuDevice};
-use cubecl::wgpu::{init_async, AutoGraphicsApi};
 
 use serde::Serialize;
 use wasm_bindgen::prelude::*;
diff --git a/examples/mnist-inference-web/Cargo.toml b/examples/mnist-inference-web/Cargo.toml
index a72b3d3c1..c8f780360 100644
--- a/examples/mnist-inference-web/Cargo.toml
+++ b/examples/mnist-inference-web/Cargo.toml
@@ -13,10 +13,11 @@ crate-type = ["cdylib"]
 default = ["ndarray"]
 
 ndarray = ["burn/ndarray"]
-wgpu = ["burn/wgpu"]
+wgpu = ["burn/wgpu", "cubecl-runtime"]
 
 [dependencies]
 burn = { path = "../../crates/burn", default-features = false }
+cubecl-runtime = { version = "0.3.0", optional = true, features = ["channel-mpsc"] } # missing feature flag
 serde = { workspace = true }
 console_error_panic_hook = { workspace = true }