spb/modelmap Public License

1---2project: modelmap3document: Phase 1 notes — §4.9 Systems side — local mapping at scale on macOS4author: Simon-Pierre Boucher5contact: contact@spboucher.ai6website: https://modelmap.io7created: 2026-08-128status: draft9---1011# §4.9 — Systems side of local interpretability at scale1213## Activation storage size math (framing)14Llama-class 7–8B (d_model 4096, 32 layers): fp16 residual capture = **8 KB/token/layer**; all layers = **256 KB/token**. MLP hidden (d_mlp 14336) is 3.5× larger per layer.15- 10M tokens × 1 layer ≈ **80 GB** — cacheable on a 1–2 TB SSD for one or two hookpoints.16- SAE budget (500M tokens × 1 layer) ≈ **4 TB fp16** — pre-caching infeasible; **on-the-fly generation mandatory**.17- Full-model capture for 100k tokens ≈ 25–90 GB — fine for probing/patching corpora (10⁴–10⁶ tokens).18**The Mac regime splits cleanly: small causal/probing corpora → cache everything; SAE-scale corpora → stream, never store.**1920## Storage formats21- **safetensors** https://github.com/safetensors/safetensors — zero-copy mmap, lazy per-tensor; no appendable writes, no chunk index; free-form metadata dict (exploit for provenance embedding).22- **zarr** https://github.com/zarr-developers/zarr-python — chunked, compressed, appendable N-D; ideal shape for token×layer×d_model with random chunk access; but dense fp16 activations are high-entropy → compression gains modest; reported slower than h5py/npy raw throughput. **Benchmark on macOS, don't assume.**23- **HDF5** — mature; single-writer pain; little advantage over zarr/mmap here.24- **Raw mmap / np.memmap** — best random-access latency (the pattern SAE shuffling needs); on unified memory + Apple NVMe likely the throughput winner; needs external index + own content hashes. **No published macOS numbers exist for any of these — Experiment H claims this gap.**2526## SAE training pipelines (streaming on limited memory)27- **SAELens** https://github.com/decoderesearch/SAELens — on-the-fly buffer shuffle (approximate); optional CacheActivationsRunner (safetensors shards); buffer size = the RAM knob; no provenance schema for caches; **shuffle quality is a replication variable modelmap should control.**28- **EleutherAI sparsify** https://github.com/EleutherAI/sparsify — TopK SAEs/transcoders **with no activation caching at all**; bitsandbytes 8-bit path is CUDA-only (useless on MPS).29- **Anthropic (disclosed):** collect billions → **full on-disk shuffle** → stream (https://transformer-circuits.pub/2024/scaling-monosemanticity/). The gold standard a Mac cannot replicate. **Open testable question: how much does buffer-shuffled streaming degrade feature replication vs full shuffles? Testable locally at 0.5–3B.**30- Budget: (weights + KV cache + buffer + SAE) < unified memory. 3B fp16 + 2 GB buffer + 131k SAE fits 16 GB; 8B needs 32 GB+.3132## Hooks and capture paths33- **PyTorch MPS:** forward hooks fire normally (eager); costs: forced syncs on `.cpu()` copies, operator gaps (`PYTORCH_ENABLE_MPS_FALLBACK=1` silently round-trips via CPU), no float64, **`device_map="auto"` CPU offload does not work on MPS — model must fit unified memory**. fp16 numerics on MPS can differ from CUDA — a confound when comparing to published results. Hook overhead unmeasured in literature (Experiment H).34- **TransformerLens** https://github.com/TransformerLensOrg/TransformerLens — HookPoints everywhere; MPS opt-in; materializes all hook activations eagerly (memory-hungry); HF checkpoints only, no GGUF.35- **MLX / mlx_lm:** lazy evaluation (https://ml-explore.github.io/mlx/build/html/usage/lazy_evaluation.html) — intermediates can be *retained for free* during graph construction; early materialization wrecks throughput; **no built-in hook API — introspection = wrapping modules** (plain Python, easy). MLX runs 4-bit quantized models natively → **the only realistic path to quantized-model activation capture with a Python API** (llama.cpp aside). M5 neural-accelerator work: https://machinelearning.apple.com/research/exploring-llms-mlx-m536- **mlxterp** https://github.com/coairesearch/mlxterp — the one existing MLX mechinterp library: `model.trace()` context manager, ~196 activations/forward, composable interventions, mlx-lm model support. **Very early (~11 stars); no streaming store, no SAE training, no provenance, no overhead benchmarks. Direct prior art for modelmap's capture layer — evaluate before building.**3738## Quantized-runtime introspection (llama.cpp)39- **eval-callback** https://github.com/ggml-org/llama.cpp/tree/master/examples/eval-callback — ggml graph callback intercepting every op; can dump every intermediate tensor during inference **including the Metal backend** — real C-level activation extraction from *actually-quantized* compute.40- **imatrix** — per-channel mean-squared activation statistics guiding quantization; itself a crude weight-importance map.41- **Verified absent: no interpretability tooling wraps this.** A thin exporter (eval-callback → mmap store) would give Q8/Q4/Q2 activation capture no Python framework offers — **key for Experiment F.** Caveats: dequantized dumps op-by-op have real overhead; no token bookkeeping; graph names shift across versions — pin a commit.4243## Metal capture & profiling44- `MTLCaptureManager` (https://developer.apple.com/documentation/metal/mtlcapturemanager) records `.gputrace` (30k+ buffers per forward) — a debugging tool, not a data pipeline. Instruments' Metal System Trace for kernel timings/occupancy → **Experiment H instrumentation**, and for validating that Python-level capture matches GPU reality.4546## The competing philosophy: NNsight + NDIF47- https://arxiv.org/abs/2407.14561 (ICLR 2025); https://ndif.us/ ; European replication eDIF: https://arxiv.org/pdf/2508.1055348- Deferred-execution intervention graphs shipped to a shared GPU fabric — *don't run locally; send the experiment away*. **modelmap's antithesis and best contrast case:** NDIF serves unquantized checkpoints and cannot study the quantized runtimes people actually use locally; reproducibility depends on a remote service. NNsight's deferred-graph idiom is conceptually close to MLX laziness — study for API design.4950## Disk/IO on macOS51- APFS: native sparse files (pre-sized mmap stores written out of order allocate nothing); **copy-on-write clones = instant snapshot of an activation store — a genuinely nice provenance primitive**; metadata-only checksums (data corruption NOT detected — store own content hashes). Refs: https://eclecticlight.co/2024/06/08/apfs-how-sparse-files-work/ ; https://eclecticlight.co/2021/03/29/sparse-files-are-common-in-apfs/52- fio-style numbers for mmap workloads on Apple NVMe essentially unpublished — **measure, don't import Linux ext4 folklore.** (Sister project localvm-research already measured cold-cache NVMe: ~13.1 GB/s ceiling at ≥256 KiB blocks, QD≥4 — reuse.)5354## Verified open gaps (modelmap's to claim)551. No published MLX-vs-MPS hook-overhead benchmarks.562. No macOS activation-store format benchmarks (mmap vs zarr vs safetensors on APFS).573. No quantized-runtime interpretability tooling (llama.cpp eval-callback unwrapped).584. Buffer-shuffle vs full-shuffle effect on SAE feature replication unpublished.595. 2026 systems-side interest appearing (https://arxiv.org/pdf/2605.11093 — model-internal observability for LLM inference) — move fast.60