Skip to main content

Module probe

Module probe 

Source
Expand description

Probe inference helpers for startup measurement and arena priming.

Functionsยง

probe_run_dense ๐Ÿ”’
Runs a single session.run() for the probe, measuring RSS before and after.
run_probe_batch ๐Ÿ”’
Runs one probe batch: tokenize texts, build padded arrays, call session.run(), and return RSS deltas. Uses embed_denseโ€™s no-pad tokenizer path.