oxicuda-lm 0.1.6

Large language model inference primitives for OxiCUDA: BPE tokenizer, transformer layers with KV cache, GPT-2 and LLaMA architectures — pure Rust, zero CUDA SDK dependency.
Documentation 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23

//! BPE tokenizer and vocabulary management.
//!
//! # Usage
//!
//! ```ignore
//! use oxicuda_lm::tokenizer::{BpeBuilder, BpeTokenizer};
//!
//! let tokenizer = BpeBuilder::new()
//!     .add_merge(b"a", b"b")   // "ab" → id 256
//!     .add_special("<eos>", 0)
//!     .build()?;
//!
//! let ids = tokenizer.encode("ab")?;
//! assert_eq!(ids, vec![256]);
//! let text = tokenizer.decode(&ids)?;
//! assert_eq!(text, "ab");
//! ```

pub mod bpe;
pub mod vocab;

pub use bpe::{BpeBuilder, BpeTokenizer};
pub use vocab::Vocab;