Add initial ML implementation

Signed-off-by: Nikolaos Karaolidis <nick@karaolidis.com>
This commit is contained in:
2024-03-29 13:40:49 +00:00
parent f715881b07
commit d7e9350257
26 changed files with 4441 additions and 104 deletions

View File

@@ -11,6 +11,10 @@ path = "src/lib/qrust/mod.rs"
name = "qrust"
path = "src/bin/qrust/mod.rs"
[[bin]]
name = "trainer"
path = "src/bin/trainer/mod.rs"
[profile.release]
panic = 'abort'
strip = true
@@ -70,7 +74,14 @@ lazy_static = "1.4.0"
nonempty = { version = "0.10.0", features = [
"serialize",
] }
rand = "0.8.5"
rayon = "1.9.0"
burn = { version = "0.12.1", features = [
"wgpu",
"cuda",
"tui",
"train",
] }
[dev-dependencies]
claims = "0.7.1"
serde_test = "1.0.176"