feat: migrate to pure Rust backend with WGPU/Vulkan support

- Remove libtorch dependency and related features
- Switch to ndarray backend as default for CPU execution
- Update README to reflect new WGPU/Vulkan backend usage
- Simplify device selection to use only CPU backend
- Enable WGPU backend via feature flag for GPU acceleration
This commit is contained in:
2026-03-03 22:07:14 +01:00
parent 6cfd6db5a5
commit 754810ca88
3 changed files with 11 additions and 36 deletions

View File

@@ -7,14 +7,12 @@ edition = "2021"
[features]
wgpu-backend = ["burn-wgpu"]
default = ["wgpu-backend"]
[dependencies]
burn = "0.20.1"
burn-ndarray = "0.20.1"
burn-tch = "0.20.1"
burn-autodiff = "0.20.1"
burn-wgpu = { version = "0.20.1", optional = true }
tch = "0.22.0"
serde = {version = "1.0.171", features = ["std", "derive"]}
npy = "0.4.0"
num-traits = "0.2.15"