gitwtfhub

wtf is jspace-viz?

festyve/jspace-viz — explained in plain English

Analysis updated 2026-05-18

13PythonAudience · researcherComplexity · 4/5Setup · moderate

TL;DR

A visual tool that shows, layer by layer, what a language model seems to be thinking as it reads a prompt, before it answers.

Mindmap

mindmap
  root((jspace-viz))
    What it does
      Layer by layer readout
      Live typing mode
      Lens comparison
    Tech stack
      Python
      HuggingFace models
      Jacobian lens
    Use cases
      Model interpretability research
      Comparing lens methods
      Exploring model internals
    Audience
      Researchers
      ML enthusiasts

Code map

Detail Auto

An interactive map of this repo's files and how they connect — its source is parsed live in your browser. Click Visualize to build it.

filefunction / class

Why would anyone build with this?

REASON 1

Explore what a language model is internally thinking as it reads a prompt.

REASON 2

Compare the Jacobian lens against the classic logit lens on the same model.

REASON 3

Fit a new interpretability lens to a model that does not have one yet.

What's in the stack?

PythonPyTorchHuggingFaceNumPy

How it stacks up

festyve/jspace-viz1lystore/awaek47cid/wp2shell-lab
Stars131313
LanguagePythonPythonPython
Setup difficultymoderatemoderatemoderate
Complexity4/52/54/5
Audienceresearchervibe coderresearcher

Figures from each repo's GitHub metadata at analysis time.

How do you spin it up?

Difficulty · moderate Time to first run · 30min

Larger model presets need a machine with at least 16GB of RAM or a GPU.

Wtf does this do

jspace-viz is a visual tool for looking inside open weights language models while they process text. It lets you type any prompt and watch, layer by layer, what the model seems to be thinking at each word position, before it produces its final answer. There is also a free online demo you can try in your browser without installing anything. The tool is built on an idea from a 2026 research paper about how language models may hold a shared, readable set of concepts across their internal layers, something the paper calls a global workspace. To make those inner layers readable, the tool uses a technique called the Jacobian lens, which translates the model's internal signals into words using a mathematical transformation learned from example text. This is more accurate than an older method called the logit lens, especially for early and middle layers of the model. The tool lets you switch between the two methods to compare them side by side. When you type a prompt, the screen shows a grid where each column is one word or token of your input and each row is one layer of the model. Each cell in the grid shows what word the model would say if it had to answer right at that point, colored to reflect how confident it is. A separate panel highlights concepts the model appears to be holding in mind that are not simply words copied from the prompt or the next word it will say. Clicking any cell shows a detailed ranking of possibilities, and charts track how word like or concept like the model's internal state is at each layer. The project works with several existing language models, including small ones like GPT-2 and larger ones such as DeepSeek and Llama variants, and it includes scripts for fitting the required lens data to a new model on a personal computer. It is aimed at researchers and technically curious people who want to explore how language models process information internally, and setup requires some comfort with Python and command line tools.

Yoink these prompts

Prompt 1
Help me set up jspace-viz locally and run the gpt2 preset to explore a prompt.
Prompt 2
Explain how the Jacobian lens in jspace-viz differs from the classic logit lens.
Prompt 3
Walk me through fitting a new lens for a model using jspace-viz's fit_lens.py script.
Prompt 4
Show me how to read the workspace panel and rank heatmap in jspace-viz.

Frequently asked questions

wtf is jspace-viz?

A visual tool that shows, layer by layer, what a language model seems to be thinking as it reads a prompt, before it answers.

What language is jspace-viz written in?

Mainly Python. The stack also includes Python, PyTorch, HuggingFace.

How hard is jspace-viz to set up?

Setup difficulty is rated moderate, with roughly 30min to a first successful run.

Who is jspace-viz for?

Mainly researcher.

View the repo → Decode another repo

This repo across BitVibe Labs

Don't trust strangers blindly. Verify against the repo.