high-performance bpe tokenization with rust core
Implements Byte Pair Encoding (BPE) algorithm in Rust with FFI bindings to Python and Node.js, achieving 10-100x faster tokenization than pure Python implementations. The Rust core uses efficient data structures and memory management to process text into token IDs and offsets, with the tokenization pipeline flowing through normalizers, pre-tokenizers, and post-processors as composable stages.
Unique: Single Rust implementation compiled to Python (PyO3) and Node.js (napi-rs) bindings ensures byte-identical tokenization across languages; Rust core eliminates GIL contention and enables true parallelization via Arc<RwLock> thread-safe wrappers, unlike NLTK/spaCy which are Python-first
vs alternatives: 10-100x faster than pure Python tokenizers (NLTK, spaCy) and maintains consistency across Python/Node.js/Rust, whereas SentencePiece is C++ only and requires separate Python wrapper maintenance
wordpiece tokenization with subword vocabulary matching
Implements WordPiece algorithm (used by BERT, DistilBERT) that greedily matches the longest subword tokens from a vocabulary, prefixing continuation tokens with '##' to indicate non-initial positions. The algorithm processes pre-tokenized words character-by-character, falling back to [UNK] tokens for out-of-vocabulary subwords, enabling efficient representation of rare words and morphological variants.
Unique: Implements greedy longest-match WordPiece with configurable [UNK] token fallback and ## continuation markers; supports both training from corpus and loading pre-trained vocabularies, unlike NLTK which lacks WordPiece entirely
vs alternatives: More efficient than BPE for morphologically rich languages and better preserves semantic units than character-level tokenization, though less flexible than SentencePiece's unigram language model approach
multi-language binding support with pyo3 (python) and napi-rs (node.js)
Provides language-specific bindings that expose the Rust core to Python and Node.js via PyO3 and napi-rs FFI technologies. PyO3 bindings use Arc<RwLock> for thread-safe shared state and integrate with tokio for async support; napi-rs bindings compile to native addons for multiple platforms (Linux gnu/musl, Windows, macOS, Android). Both bindings maintain API parity with the Rust core while providing idiomatic interfaces for each language.
Unique: Single Rust implementation compiled to idiomatic Python (PyO3 with Arc<RwLock> thread safety) and Node.js (napi-rs native addons) bindings, ensuring byte-identical tokenization across languages; PyO3 integration with tokio enables async tokenization without GIL
vs alternatives: More consistent across languages than separate implementations (SentencePiece C++ + Python wrapper) and better performance than pure Python (NLTK, spaCy); comparable to transformers library but with more explicit language binding architecture
batch tokenization with parallel processing support
Supports efficient batch tokenization of multiple texts simultaneously, with optional parallelization across CPU cores. The batch API accepts lists of strings and returns lists of Encoding objects, with internal parallelization via Rayon (Rust) or thread pools. Batch processing reduces per-text overhead and enables better CPU cache utilization compared to sequential tokenization.
Unique: Implements batch tokenization with automatic Rayon-based parallelization in Rust core, reducing per-text overhead and enabling efficient multi-core utilization; batch API is exposed to Python/Node.js with configurable thread pool size
vs alternatives: More efficient than sequential tokenization loops (2-4x speedup on 8-core systems) and simpler than manual threading (no GIL contention in Python); comparable to transformers library's batch_encode_plus but with more transparent parallelization
encoding object with rich metadata and token-level information
Returns Encoding objects that encapsulate complete tokenization results: token IDs, token strings, character offsets, attention masks, token type IDs (for sequence pairs), and special token positions. The Encoding structure provides convenient accessors for common operations (e.g., getting tokens for a span, padding to length) and supports serialization to/from dictionaries for integration with ML frameworks.
Unique: Provides a rich Encoding object that captures complete tokenization state (token IDs, strings, offsets, masks, token type IDs) with convenient accessors for common operations; supports padding/truncation with automatic mask updates and serialization to/from dictionaries
vs alternatives: More comprehensive than raw token ID arrays (includes offsets, masks, token type IDs) and more convenient than separate token/offset lists; comparable to transformers library's BatchEncoding but with more explicit metadata structure
decoder for reconstructing text from tokens
Implements decoders that reconstruct original text from token sequences, reversing the tokenization process. Different decoders handle different tokenization schemes: BPE decoder removes ## markers and merges subword tokens, WordPiece decoder handles ## continuation markers, Unigram decoder reconstructs from byte-level tokens. Decoders support optional space insertion and special character handling.
Unique: Provides algorithm-specific decoders (BPE, WordPiece, Unigram) that reverse tokenization by removing subword markers and merging tokens; supports optional space insertion and special character handling for different languages
vs alternatives: More accurate than naive token concatenation (handles ## markers and byte-level tokens) and simpler than custom decoding logic; comparable to transformers library's decode methods but with more explicit decoder selection
unigram language model tokenization with probability-based selection
Implements Unigram tokenization (used by SentencePiece) that models tokenization as a probabilistic process where each token has an associated loss value. During encoding, the algorithm finds the most likely tokenization sequence that minimizes loss, and during training, iteratively removes low-loss tokens from the vocabulary. This approach naturally handles variable-length tokens and rare characters without explicit [UNK] fallback.
Unique: Uses probabilistic loss-based token selection instead of greedy matching, enabling graceful handling of unknown characters through byte-level fallback without [UNK] tokens; EM-based training iteratively optimizes vocabulary for corpus-specific loss minimization
vs alternatives: Better multilingual support than WordPiece (no language-specific preprocessing needed) and more principled than BPE (probability-based vs heuristic merge frequency), though slower than BPE at inference time
wordlevel tokenization with simple vocabulary lookup
Implements the simplest tokenization strategy: direct vocabulary lookup where each whitespace-separated word maps to a token ID, with [UNK] for out-of-vocabulary words. This approach requires explicit pre-tokenization and is primarily used for legacy models or as a baseline, but provides maximum interpretability and minimal computational overhead.
Unique: Provides the minimal tokenization implementation for compatibility and interpretability; no subword decomposition or probabilistic selection, just direct vocabulary lookup with [UNK] fallback
vs alternatives: Simpler and more interpretable than BPE/WordPiece/Unigram for debugging, but unsuitable for production NLP due to high OOV rates and poor morphological handling
+6 more capabilities