embedding 0.1.4

A Rust library and CLI for training embeddings from scratch
Documentation 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55

[package]
name = "embedding"
version = "0.1.4"
edition = "2024"
description = "A Rust library and CLI for training embeddings from scratch"
authors = ["Ying Kit WONG"]
license = "Apache-2.0"
repository = "https://github.com/yingkitw/embedding"
keywords = ["embedding", "machine-learning", "nlp", "vector", "ai"]
categories = ["science", "command-line-utilities"]
documentation = "https://docs.rs/embedding"
homepage = "https://github.com/yingkitw/embedding"

[lib]
name = "embedding"
path = "src/lib.rs"

[[bin]]
name = "embedding"
path = "src/main.rs"

[features]
default = []
gpu = ["dep:wgpu", "dep:pollster", "dep:bytemuck"]

[dependencies]
ndarray = "0.15"
ndarray-rand = "0.15"
serde = { version = "1.0", features = ["derive"] }
serde_json = "1.0"
clap = { version = "4.5", features = ["derive"] }
rand = "0.8"
tracing = "0.1"
tracing-subscriber = "0.3"
bincode = "1.3"
indicatif = "0.17"
prost = "0.13"
bytes = "1.10"
unicode-normalization = "0.1"
memmap2 = "0.9"
wgpu = { version = "0.20", optional = true }
pollster = { version = "0.3", optional = true }
bytemuck = { version = "1.14", optional = true }

[build-dependencies]
prost-build = "0.13"

[dev-dependencies]
criterion = "0.5"
tempfile = "3.14"
proptest = "1.5"

[[bench]]
name = "embedding_bench"
harness = false