diff --git a/Cargo.lock b/Cargo.lock
index e69bfdc1c..0e84add9d 100644
--- a/Cargo.lock
+++ b/Cargo.lock
@@ -461,22 +461,6 @@ dependencies = [
  "thiserror",
 ]
 
-[[package]]
-name = "burn-cuda"
-version = "0.14.0"
-dependencies = [
- "burn-common",
- "burn-compute",
- "burn-fusion",
- "burn-jit",
- "burn-tensor",
- "bytemuck",
- "cudarc",
- "derive-new",
- "half",
- "log",
-]
-
 [[package]]
 name = "burn-dataset"
 version = "0.14.0"
@@ -3792,6 +3776,14 @@ dependencies = [
  "thiserror",
 ]
 
+[[package]]
+name = "refactor"
+version = "0.14.0"
+dependencies = [
+ "burn",
+ "serde",
+]
+
 [[package]]
 name = "regex"
 version = "1.10.4"
diff --git a/Cargo.toml b/Cargo.toml
index a2df76a7c..49b3eb802 100644
--- a/Cargo.toml
+++ b/Cargo.toml
@@ -14,7 +14,10 @@ members = [
     "xtask",
 ]
 
-exclude = ["examples/notebook"]
+exclude = [
+    "examples/notebook",
+    "crates/burn-cuda"     # comment this line to work on burn-cuda
+]
 
 [workspace.package]
 edition = "2021"