UNesT — single-patch parity/debug probe

Random/fixed local probe for transformer forward parity. Product validation is the worker upload path with real whole-brain T1 data. NestViT (3-level hierarchical transformer) + CNN decoder, 87 M params.  ·  batched-fast worker v3  ·  outer-loop worker v2  ·  TopCoW
Parity/debug scope. Runs nestViT (patch embed → 3 nested transformer levels → final LN) followed by a U-Net-style CNN decoder with 5 skip connections, entirely via GPU-resident persistent buffers. Weight tensors — including the proj.weight row permutation fix for UNesT's non-standard attention merge — are uploaded directly from a 333 MB .safetensors file.
⚠ Download size: ~340 MB of model weights + inputs will be fetched below. First load will take a while on a slow connection.

Log

Axial slice preview (class argmax)

 
Input (random parity volume)
WASM argmax
PyTorch argmax