TrainForge

Watch the guided demo → Interactive platform is coming soon

What if AI could turn a research goal into a better model—and prove the improvement is real?

That’s what we’re building with TrainForge: an autonomous R&D system that generates ideas, implements them, runs training and evaluation, and learns from the outcome. Built by Salesforce AI Research.

Inside the loop of TrainForge

Input Baseline model + targetTask, evaluation & compute budget

Autonomous R&D

Learn from the result. Repeat.
Output Best verified checkpointModel + evaluation record

No improvement means no promotion.

Why this is new

It doesn’t just train. It decides what to try next.

TrainForge discovers training ideas, tests them, and turns every result into the next, better proposal—while verified code and checkpoint-bound evidence keep the loop under control.

Novelty

Design new strategies end to end

Existing methods are references, not a fixed menu. Agents search broadly, debate alternatives, and redesign the full improvement loop around each target. Each proposal remains a novelty candidate until prior-work checks and experiments support it.

search broadly → design end to end → test novelty
Fidelity

Make code match the idea

No repository or recipe is provided. TrainForge chooses what the codebase needs, Codex writes it from an empty workspace, and checks verify that the running code still matches the idea.

freeze intent → write from scratch → verify
Efficient exploration

Test cheaply, scale what works

Ideas start with a fast LoRA test. Only promising ones move to a short full-parameter run and then full training.

LoRA test → short full-parameter run → full training
Evolution

Improve ideas across rounds

Every result—success or failure—changes the next proposal, so the method improves across rounds.

test → learn → regenerate

Tasks and evaluations

What we test, how we test it, and what changed.

You provide the target. TrainForge chooses the research direction and trains on separate training data; the datasets below are frozen, held-out evaluations only. The effects shown are averages across the listed benchmarks.

Guided demonstration

Watch an idea improve across multiple rounds.

See the complete story in one interface: diagnose the gap, generate and challenge ideas, steer the system, verify the code, train, evaluate, and promote.

Interactive platform is coming soon

The idea-evolution lineage is illustrative. The two rejected screens and the successful Qwen3-4B reference result are real measurements and are labeled in the demo.

Idea card

Research name

01 · Failure mode

02 · Missing training signal

03 · Training intervention

Worked example

Why it is different