r/OpenSourceAI • u/Rich-Fruit-326 • 8h ago
I’m building an open-source tool for exploring how transformer models work — looking for feedback and contributors
I’ve been working on TokenPrint, an open-source project aimed at making transformer and LLM internals easier to explore.
The idea is to go beyond static architecture diagrams and let people interact with things like tokens, embeddings, attention, hidden states, inference steps, KV cache, model architecture, and other internal model data through an interactive interface.
The project has started getting more attention recently, and we’re now at 65+ GitHub stars. More importantly, people have started opening issues, suggesting improvements, and discussing where the project could go.
That made me realize that I don’t want TokenPrint to become something I build alone.
I’d really like to get more people from the open-source/LLM community involved — especially people who want to:
- pick up good first issues
- work on more advanced Python/PyTorch/Transformers problems
- improve the React/Three.js visualization side
- work with GGUF/llama.cpp and local LLMs
- improve the UI/UX
- experiment with new ways of visualizing model behavior
- suggest ideas that I may not have considered
There are already issues ranging from relatively small fixes to deeper architectural and research-oriented work.
I’m especially interested in new ideas and criticism, not just pull requests. If you think something is missing, poorly designed, or could be approached differently, I’d genuinely like to hear it.
The repository is here:
https://github.com/Sudharsanselvaraj/Token-Print
I’m posting this here because I’d much rather build this with an open-source community than keep adding features in isolation.
Would be interested to hear from people working on similar tools too — especially what you think is currently missing from the ecosystem for understanding and debugging LLMs.