Team
I've already shipped this pattern in production,
at AWS, for cloud architectures.
At AWS I led an agentic LLM that converted natural-language cloud requirements into validated AWS architecture diagrams via typed-schema function-calling. 5× faster Solutions Architect workflow.
Neurarch applies the same pattern to neural networks. Technical risk is small. Week-2 traction confirms ML engineers want it.
Meta · Research Scientist (current)
AI startup · Founding Lead Scientist · solo-shipped LLM vuln triage in 10 wks, led 5-person team
AWS AI · Applied Scientist · 5 yrs · Bedrock GenAI, agentic LLM NL→AWS-architecture system
Supporting record
granted
citations
workshop
NJIT
MVP
types
lint rules
export
Solo by design, to ship at a team's velocity (full stack in 21 days).
The Problem
Models have no compiler.
You find the bug when the GPU run dies.
PyTorch has no compiler for architecture. A shape mismatch, or a head dim that doesn't divide the embed dim, stays silent until runtime.
A shape crash fails fast. The expensive ones are silent: a missing residual, a misplaced norm, an aux-loss you forgot, train without crashing and just train badly. You find out hours in. A model takes 10 to 30 cycles before it even trains.
Architecture lives in cell 47 of a notebook, or an Excalidraw box that doesn't know it's a Conv2D. Unreadable to your team, uncomputable, un-exportable.
Not just our opinion: in an Alibaba study of 12,289 failed training jobs, tensor-shape errors were the second most common framework-specific failure. Peer-reviewed static checkers (ICSE 2022, ASE 2021) catch this class in seconds with zero false positives. We productized that finding.
Our user: the ML researcher / applied scientist who designs or forks architectures weekly. Today they live in Jupyter cells, Slack screenshots, and merge conflicts.
The Solution
One canvas. Live params. An agent that edits.
Real code on the way out.
Every layer is a first-class, typed object, not a box in a diagram. The agent, version control, and refactor work like a real IDE. And it knows BatchNorm wants a Conv before it, because it was built for ML, not retrofitted into it.
60-Second Demo
Idea → live model → deployable code.
In one minute.
"Build me a CLIP-style dual encoder, ViT-B/32 image, BERT-base text."
42 layers across two encoders + contrastive head. Live: 219M params · 4.6 GFLOPs · A100 fit ✓.
"Swap ViT-B → ViT-S to fit on a T4." Canvas re-edits live: 88M params · 1.4 GFLOPs.
/python → clip_dual.py (PyTorch). Auto-generated tests: 4 passed in 1.8s.
/cloud → SkyPilot YAML + Modal training job + Dockerfile. Live in your cluster.
Same loop replaces 3 to 5 days of Jupyter debugging, Slack threads, and "does this even compile?" stress.
Product
A typed graph of your model. An agent that edits it.
- ›135+ specialized panels
MACs flame chart · loss landscape · CKA similarity · attention chord · neuron activity - ›Multi-provider AI agent on a typed graph
Gemini · Claude · GPT-4 · Ollama, graph-grounded context, applies diffs in place - ›Look inside any layer
Layer Inspector renders any of 182 layer types' real computation offline, then overlays trained weight stats + per-epoch evolution after a run. - ›Frontier-scale, paper-readable
Level-of-detail folding collapses a 61-layer DeepSeek-V3 or LLaMA-3 into "× N" blocks, the way papers draw them. - ›Paper figure → runnable model
Drop a diagram screenshot, a vision LLM rebuilds nodes + residual edges. - ›Own-your-code import & export · 28 SOTA templates
Residuals, names, provenance preserved; PyTorch / Keras / TF / JAX / ONNX / Caffe / MXNet + FastAPI + Docker.
$ pytest test_resnet50.py → 4 passed in 1.8s
Why Now
The architect bottleneck just got 10× wider.
PMs and designers now want to architect models. Code-only tools shut them out. A graph + agent loop opens the gate.
Companies fork & fine-tune dozens of models per quarter. Without structural diff & provenance, knowledge evaporates.
Agents that "write architectures" produce 600-line .py files no one trusts. Typed structure makes review possible.
Market Size
Every 1% of seats = $2.2M ARR.
600K addressable today · $1.5B+ TAM over 5 years.
33K GitHub stars on Netron, a read-only ML inspection tool. Stars are softer than dollars, but 33,000 engineers wanted half of what we ship badly enough to star it.
~$400M ARR (2024) on a similar per-IC-developer seat motion. Younger, faster-growing, higher-ACV category than ours.
+400K academic ML researchers, +500K Fortune-5000 data-science seats. $1.5B+ TAM by year 5.
Business Model
Freemium SaaS with a
self-served Pro tier.
- Typed-graph editor + 28 SOTA templates
- PyTorch / Keras export
- Bring-your-own LLM key
- Public sharing
- Everything in Free
- Hosted Gemini agent (500/mo)
- HuggingFace direct import
- FastAPI & cross-framework export
- Private models + version history
- Everything in Pro
- Hosted Claude agent
- Gemini 2000/mo · higher rate limits
- Modal & SkyPilot cloud GPU launcher
- All 135+ panels unlocked
- SSO, audit log, RBAC
- Real-time collaboration
- On-prem / VPC deploy
- Custom layer plugin SDK
- SLA + dedicated support
~1.3K Pro · 50 paying teams · Netron outreach + waitlist conversion
~3K Pro/Plus · 10 Team customers · seed round
~10K Pro · 30 Team / Enterprise
~50K Pro · 100+ Enterprise · plugin marketplace rev share
Stripe live (off pending pricing study) · Modal.com GPU backend integrated · Supabase auth shipped.
Traction · since launch
9 design partners, 6 in weekly dialogue.
Live users from 7 frontier ML orgs.
6 in weekly dialogue
zero paid acquisition
(7 fixes in one day, V4 paper)
"If the full paper-to-runnable-code path lands end-to-end, this tool is unbeatable."
Stripe wired and tested. Payments deliberately off for now, pending a 50-user pricing study. Flipping live = 1-day change.
Go-to-market
Open-source wedge → self-serve Pro
→ team expansion.
The free tool is the funnel. Pull before push: 9 design partners, 1 paid pilot in scoping, $0 marketing.
neurarch-lint (MIT CLI) + MCP server + open SOTA templates run in any PyTorch repo, CI, or Claude. Each bug they catch is a reason to open the hosted app.
Free web app → Pro $19 when they want the hosted agent, private models, and cross-framework export. Paywalls already live behind Stripe.
Individual seat → team. Already inside Apple, AbbVie, and other design-partner accounts; 1 paid pilot in scoping. We expand seat-by-seat in accounts we're in.
Every model designed = a labeled architecture. Exported figures and public templates carry a "Made with Neurarch" mark back to the top of funnel.
CLI → web → Pro conversion: the one metric that sets CAC
In-CLI "open in Neurarch" CTA on every caught bug
Design-partner accounts → first paid seats
OSS template drops on HF · X · r/MachineLearning
Competition
No one else ships a typed-graph contract
with an agent that respects it.
| Typed Graph | Live FLOPs | AI Agent | Code Export | Cross-FW Convert | Cloud Launch | |
|---|---|---|---|---|---|---|
| Netron | view-only | — | — | — | — | — |
| TensorBoard | post-hoc | ✓ | — | — | — | — |
| Excalidraw / Miro | — | — | — | — | — | — |
| HuggingFace Hub | — | — | — | ✓ | PT↔TF | — |
| Cursor / Copilot | — | — | ✓ | text¹ | — | — |
| MMdnn (MS, archived) | — | — | — | ✓ | 7 FW | — |
| SkyPilot | — | — | — | — | — | ✓ |
| Neurarch | ✓ | ✓ | ✓ | ✓ | 7 FW | ✓ |
Cursor writes code. Neurarch designs architecture. Complementary, but the typed-graph + agent + ML-domain combo is uncontested.
¹ Cursor edits source files; it does not export model architectures or propagate tensor shapes through a typed graph.
Positioning
The empty quadrant: design-time intelligence.
intelligence
Tools that show your model are read-only. Tools you can edit don't know it's a model. Neurarch is a typed graph you can edit.
Objection: "Doesn't HuggingFace already show model structure?"
Seeing a model isn't designing one.
HuggingFace renders a model that already exists, read-only. Neurarch sits one step upstream, where the model is still being built.
✓ Find a pre-built model
✓ Read-only structure view
✓ Hosting & inference compute
✕ Can't edit the graph
✕ No pre-training bug checks
✕ No runnable glue code
✓ An editable typed graph
✓ Catch bugs before the GPU run
dim mismatch · param blow-up · cycles
✓ Emit runnable code
train · eval · deploy
→ then push to HuggingFace to host
We're HuggingFace's on-ramp and complement, not its replacement.
Moat
Why this is hard to copy
in less than 18 months.
The durable, compounding moat. Every accepted edit, lint catch, and paper-to-graph mapping is proprietary intent → structure → outcome data. Text tools and black-box AutoML structurally cannot collect it, and it grows with every model designed. Already instrumented and visible in a live Data Loop dashboard.
The agent reads typed graph state, shapes, and lint, so its edits are structurally correct. Text wrappers (Cursor, Copilot) operate on source and cannot guarantee that.
An end-to-end loop a competitor rebuilds from scratch:
182 layer types · 135+ panels
Layer Inspector + level-of-detail folding
Own-your-code export, 7 frameworks + ONNX
Free Colab / Kaggle GPU round-trip
Vision figure → runnable model · paper ↔ code drift
Academic pipeline (TikZ + BibTeX) · deploy on 7 platforms
Roadmap
Q3 collab · Q4 plugin marketplace · 2027 enterprise.
135+ panels · multi-provider agent · Layer Inspector · level-of-detail folding · 7-FW export · GPU training (Modal + free Colab/Kaggle) · ONNX · paper-figure import · Canvas↔Code merge · FastAPI · Stripe.
incl. AI startup CEO · NJIT CS PhD · AbbVie Biostat Mgr · Apple SWE, <24h feedback-to-ship. Payments flip live after 50-user pricing study. Target: first $5K MRR by end of Q3.
WebSocket collab backend already implemented, needs ops/latency QA. LoRA loops on Llama/Mistral/Phi on existing Modal trainer. Unlocks Team tier.
Custom-layer SDK · panel SDK · paid plugins (rev share) · community templates.
First 5 design-partner enterprises ($60K+ ACV). VPC + air-gap deploy.
The typed graph for neural networks.
A typed-graph contract for ML architectures, with an agent that respects it.
The long-term bet: as every model designed here becomes labeled architecture data, Neurarch becomes the layer that predicts a model's cost and quality before a single GPU runs.
Built solo · 21 days · while at Meta · 9 design partners, 18 signups, 0 paid acquisition.
neurarch.com / pitch
Live product demo · investor data room available on request.
Xin Gao · xin.gao.njit@gmail.com