diff --git a/project/Cargo.toml b/project/Cargo.toml
index 3a7e011..ddcc56e 100644
--- a/project/Cargo.toml
+++ b/project/Cargo.toml
@@ -4,3 +4,4 @@ version = "0.1.0"
 edition = "2021"
 
 [dependencies]
+burn = { version = "0.16.0", features = ["webgpu", "train", "vision"] }