Use wgpu by default and ndarray for convert

Cargo.toml

@@ -6,7 +6,7 @@ edition = "2021"
 # See more keys and their definitions at https://doc.rust-lang.org/cargo/reference/manifest.html
 
 [features]
-default = ["torch-backend"]
+default = ["wgpu-backend"]
 torch-backend = ["burn-tch"]
 wgpu-backend = ["burn-wgpu"]
 
@@ -22,6 +22,7 @@ optional = true
 
 [dependencies]
 burn = { git = "https://github.com/burn-rs/burn.git" }
+burn-ndarray = { package = "burn-ndarray", git = "https://github.com/burn-rs/burn.git" }
 serde = {version = "1.0.171", features = ["std", "derive"]}
 npy = "0.4.0"
 num-traits = "0.2.15"