jax-js model chat

Chat LFM2.5 350M

Running locally with jax-js + WebGPU

Options

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.

Talk to an LLM

The first message downloads and caches a 676 MB fp16 checkpoint. Everything runs locally in your browser.