Turns paper sections and experiment results into conference-ready figures without the usual matplotlib hell. You paste in your methods paragraph or a results table, and it extracts what needs to be visualized. For architecture diagrams it generates them via Gemini in one of three visual styles (hand-drawn sketch, modern minimal, or icon-rich). For data it auto-picks chart types and renders with matplotlib/seaborn using proper academic styling. The style guide is opinionated, which is good because most ML paper figures look identical. Handles the tedious stuff like consistent colors, readable fonts, and making "your method" stand out in comparisons. Built for NeurIPS/ICML submissions where figure quality actually matters.

skills.sh installs
1.3K
GitHub stars
14.4K
last updated
Aug 21, 2026

Install

$ npx -y skills add orchestra-research/ai-research-skills --skill academic-plotting --agent claude-code

Installs into .claude/skills of the current project.

Files

Academic Plotting for ML Papers

Generate publication-quality figures for ML/AI conference papers. Two distinct workflows:

  1. Diagram figures (architecture, system design, workflows, pipelines) — AI image generation via Gemini
  2. Data figures (line charts, bar charts, scatter plots, heatmaps, ablations) — matplotlib/seaborn

When to Use Which Workflow

Figure TypeToolWhy
Architecture / system diagramGemini (Workflow 1)Complex spatial layouts with boxes, arrows, labels
Workflow / pipeline / lifecycleGemini (Workflow 1)Multi-step processes with connections
Bar chart, line plot, scattermatplotlib (Workflow 2)Precise numerical data, reproducible
Heatmap, confusion matrixmatplotlib/seaborn (Workflow 2)Structured grid data
Ablation table as chartmatplotlib (Workflow 2)Grouped bars or line comparisons
Pie / donut chartmatplotlib (Workflow 2)Proportional data (use sparingly in ML papers)
Training curvesmatplotlib (Workflow 2)Loss/accuracy over steps/epochs

Sponsors

Details

Repository
github.com/orchestra-research/ai-research-skills
Categories
Code Review & Quality, AI & Agent Building, Data Science & ML
First seen
Jun 3, 2026

More from orchestra-research/ai-research-skills

All 95
1mindrally avatarcssmindrally/skillsBest practices for CSS development including modern layout techniques, naming conventions, theming, and maintainable stylesheet architecture1.2Kskills.sh installs
2davila7 avatarscientific-schematicsdavila7/claude-code-templatesCreate publication-quality scientific diagrams using Nano Banana Pro AI with smart iterative refinement. Uses Gemini 3 Pro for quality review. Only regenerates 1.2Kskills.sh installs
3neo4j-contrib avatarneo4j-modeling-skillneo4j-contrib/neo4j-skillsDesign, review, and refactor Neo4j graph data models. Use when choosing node labels vs relationship types vs properties, migrating relational/document schemas t1.2Kskills.sh installs
4pproenca avatarnextjspproenca/dot-skillsNext.js 16 App Router performance, caching, server components, server actions, routing, and codebase-hygiene best practices — plus a category-major review/refac1.2Kskills.sh installs
5davila7 avatarsenior-fullstackdavila7/claude-code-templatesComprehensive fullstack development skill for building complete web applications with React, Next.js, Node.js, GraphQL, and PostgreSQL. Includes project scaffol1.2Kskills.sh installs
6sanyuan0704 avatarskill-forgesanyuan0704/code-review-expertskill forge1.2Kskills.sh installs