TeaLeaves
Visualize and measure how models attend to prompts
Instructs the agent to act as a mechanistic interpretability researcher that interprets attention data, validates prompt changes against cooking curves and logit lens metrics, and follows the pipeline's technical constraints when running experiments.
All Skills in This Repository (1)
Pure Emerald Level IndicatorsFrequently Asked Questions
FAQPage SchemaHow to install TeaLeaves?โผ
Run `npx skills add taylorsatula/TeaLeaves --all -g -y` in your terminal to install the skill globally.
What does TeaLeaves measure?โผ
It captures attention weights at every layer of a model's forward pass, showing exactly which regions of your prompt the model focuses on and when it commits to output tokens.
How does TeaLeaves improve prompt engineering?โผ
Instead of eyeballing outputs, you compare attention heatmaps and cooking curves before and after each prompt change to see measurably what improved or degraded.
Which models does TeaLeaves support?โผ
It works with any HuggingFace transformer model, auto-discovering the architecture, layer count, and attention modules from the model config.
Do I need a GPU to use TeaLeaves?โผ
Yes for the attention capture step, which runs on a remote GPU box such as Vast.ai; rendering and analysis of results run locally on any machine.
Related Repositories in Data & Analytics
View All in Data & AnalyticsโPaddleOCR
Extract text, tables, and formulas from PDFs and images
Scrapling
Scrape any website and bypass anti-bot protection with AI
last30days-skill
Research any topic across Reddit, X, YouTube, and the web