Your indexing pipeline tokenizes 10M docs in 18 hours. What single change gives the biggest speedup?

MCQ·Medium·4.0 · 0·~1 min·Relevant atDatabricks Meta

Attempt it

The right answer is **A: switch to the fast Rust-backed tokenizer with batch encoding**. HuggingFace ships two tokenizer implementations for most models: a Python reference (slow) and a Rust-backed version (fast). The fast version is built on the `tokenizers` library and runs the BPE merge logic in Rust with internal parallelism across CPU cores. Loading the fast version is one flag: `AutoTokenizer.from_pretrained(checkpoint, use_fast=True)` (this is the default for most models in 2026). The second half of the speedup is batch encoding. Instead of calling `tokenizer(text)` in a Python loop, you call `tokenizer(list_of_texts, padding=False, truncation=True)` once with the whole batch. This amortizes the Python-to-Rust call overhead and lets the Rust side parallelize across the batch. Typical speedup combining both: 20 to 100x on a workload like this. An 18-hour job often drops to 15-60 minutes. **Why the other options are wrong.** **B (GPU):** Tokenization is CPU-bound regex and BPE work, not the kind of matrix math GPUs accelerate. Moving to GPU adds transfer overhead and no compute benefit. **C (higher max_length):** Truncation is not the bottleneck. max_length controls how many tokens to keep, not how fast the tokenizer runs. **D (cache to disk):** Useful as a secondary optimization for reruns, but does not speed up the first run and does not fix the underlying slow loop.

**A: fast Rust-backed tokenizer + batch encoding.** HuggingFace ships two tokenizer implementations per model: a Python reference and a Rust-backed fast tokenizer (the `tokenizers` library). The fast version is the default for most models in 2026 (`use_fast=True`). It runs the BPE merge logic in Rust with internal CPU-core parallelism, and exposes a `__call__` that accepts either a single string or a list. The combined fix: `tokenizer(list_of_texts, padding=False, truncation=True, return_tensors=None)`. Batch encoding amortizes the Python-to-Rust boundary call overhead, and the Rust side parallelizes across cores. Typical speedup on this workload is 20-100x; an 18-hour job often drops to 15-60 minutes. **Secondary wins** worth knowing. - Drop padding and tensor conversion at tokenize time. Pad only at training/inference batch time inside the collator. - For multi-machine fan-out, use a process pool (joblib, ray, multiprocessing) to split the document set across machines; each machine still uses the fast tokenizer plus batch encoding. - For OpenAI-compatible workloads, `tiktoken` is another Rust-backed option with similar performance characteristics. **Why the wrong options are wrong.** **B (GPU):** Tokenization is byte-level regex split plus BPE merge lookup. Both are pointer-chasing CPU work, not tensor math. GPU adds nothing and adds transfer cost. **C (higher max_length):** Affects how much you keep per document, not throughput. **D (cache to disk):** Useful for re-runs (the second time the same corpus is tokenized) but does nothing for the first run. Also adds disk-write throughput as a new bottleneck.

Quick reference

TL;DR

Use the Rust-backed fast tokenizer with batch encoding (tokenizer(list_of_texts)) instead of a per-doc Python loop. CPU-parallel internally, FFI overhead amortized. Typical 20-100x speedup.

Memory aid

Easy to grasp

Imagine you have a million letters to stamp. The slow way is to walk one letter to the stamping desk, stamp it, walk back, pick up the next one. The fast way is to dump all million letters on the desk and use a machine that stamps them in parallel. The slow Python tokenizer takes a one at a time approach. The Rust fast tokenizer is the parallel-stamping machine. Batch encoding is dumping all the letters at once instead of one at a time. Same model, same output, dramatically less wasted time at the door of the stamping room. Moving to a GPU is wrong because the stamping machine is a CPU machine. Caching the stamped letters helps next time but does not fix the slow loop now. Setting a higher length cap changes how you cut the letters, not how fast you stamp them.

Key concepts

Fast tokenizer Batch encoding HuggingFace tokenizers

Detailed answer & concept explanation

~7 min read

Everything you need to truly understand this topic: intuition, mechanics, step by step explanation, code, formulas, and worked example. Click to expand.

Everything you need to truly understand this topic: intuition, mechanics, step by step explanation, code, formulas, and worked example.

Quick scan · 2–4 min

Everything important, quickly.

Open with the framing that tokenization is CPU-bound Python-call-overhead, not a model or algorithm issue. Name the two fixes that compose: use_fast=True for the Rust-backed tokenizer (internal CPU parallelism), batch encoding to amortize FFI overhead. Quote the typical 20-100x speedup. Dismantle the wrong options: GPU does not help on regex+BPE-lookup work, max_length affects cut not throughput, disk caching is a secondary win. Close with multi-machine fan-out as the next lever for truly large corpora.

Code example

python

from transformers import AutoTokenizer

# Slow: per-document loop with slow tokenizer
tok_slow = AutoTokenizer.from_pretrained('meta-llama/Meta-Llama-3-8B', use_fast=False)
results_slow = [tok_slow.encode(doc) for doc in docs]  # 18 hours on 10M docs

# Fast: Rust tokenizer + batch encoding
tok_fast = AutoTokenizer.from_pretrained('meta-llama/Meta-Llama-3-8B', use_fast=True)
results_fast = tok_fast(
    docs,  # full list, not loop
    padding=False,
    truncation=True,
    max_length=2048,
    return_tensors=None,
)['input_ids']
# 15-60 minutes on 10M docs, 20-100x speedup

# Even better for very large corpora: batch in chunks of 1K-10K
def batched(seq, n):
    for i in range(0, len(seq), n):
        yield seq[i:i+n]
results = []
for batch in batched(docs, 5000):
    results.extend(tok_fast(batch, padding=False, truncation=True)['input_ids'])

Real-world examples

Real products, models, and research that use this idea.

HuggingFace tokenizers library ships Rust-backed fast tokenizers for Llama 3+, Mistral, Qwen 3.5, and most modern open-weight models; use_fast=True is the default in transformers 4.40+.
tiktoken (OpenAI's Rust-backed BPE library) achieves similar throughput characteristics for cl100k_base and o200k_base workloads.
Ray Data and Dask DataFrames provide multi-machine fan-out patterns where each worker runs a fast tokenizer on a document partition.
LlamaIndex and Haystack indexing pipelines for RAG over 10M+ document corpora explicitly recommend the fast-tokenizer + batch pattern as the baseline configuration.

Interviewer follow-ups

What an interviewer would ask next. Try answering before peeking at the approach.

QWhen does the fast tokenizer NOT provide a 20-100x speedup?

QYou batch in chunks of 5,000 and the process hits OOM. What is going on and how do you fix it?

QWhy would you ever want to use the slow Python tokenizer in production?

Don't say this

Red flags and common mistakes that signal junior thinking. Click to expand.

Most common mistake

Assuming tokenization performance is a GPU problem or an algorithm problem, when the real issue is Python call overhead on a slow per-document loop.

60-second cheat sheet

The night-before-the-interview bullets. Scan these on the way to the call.

References

Primary sources. Skim if you want the original framing.