Running locally with jax-js + WebGPU
Switching models keeps each checkpoint cached in your browser.
WebGPU uses fp16 weights. Wasm casts weights to fp32 on load.
KV cache is allocated dynamically for the current chat.
The first message downloads and caches a 676 MB fp16 checkpoint. Everything runs locally in your browser.