added temp enc
Some checks failed
/ Tests (push) Failing after 5m16s

This commit is contained in:
2026-04-02 16:58:07 +05:30
parent 3ff3e88f75
commit de3f3990ee
10 changed files with 186 additions and 19 deletions

View File

@@ -4,7 +4,8 @@ version = "0.1.0"
edition = "2024"
[dependencies]
burn = { version = "0.20.1", features = ["wgpu", "std", "fusion", "ndarray"] }
burn = { version = "0.20.1", features = ["wgpu", "std", "fusion", "ndarray", "cuda"] }
burn-autodiff = "0.20.1"
[dev-dependencies]
rstest = "0.26.1"