token-sampling 0.1.0

Composable token sampling over f32 logits with caller-supplied randomness
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

[package]
name = "token-sampling"
version = "0.1.0"
edition = "2021"
rust-version = "1.63"
authors = ["Justin Jones <hey.jones@icloud.com>"]
license = "MIT"
description = "Composable token sampling over f32 logits with caller-supplied randomness"
keywords = ["sampling", "logits", "decoding", "probability", "language-model"]
categories = ["algorithms", "science"]
readme = "README.md"
repository = "https://github.com/hey-jj/token-sampling"
documentation = "https://docs.rs/token-sampling"
include = [
    "src/**",
    "tests/**",
    "Cargo.toml",
    "Cargo.lock",
    "README.md",
    "CHANGELOG.md",
    "LICENSE",
]

[features]
default = []
rand_core = ["dep:rand_core"]

[dependencies]
rand_core = { version = "0.6.4", optional = true }

[dev-dependencies]
serde_json = "1.0.151"
stats_alloc = "0.1.10"