Founding ML Research Engineer

About the Role

Join the founding team at a Y Combinator-backed AI assistant startup to build the models and systems behind a useful, reliable AI agent. You will lead hands-on research and engineering that improves answer quality while managing latency, cost, and safety, and ship improvements to users regularly.

What You'll Do

Build evaluation harnesses using model judges, human preference data, and regression tests to guide release decisions.

Post-train models through supervised fine-tuning, preference tuning, and distillation, then ship the results to production.

Develop model routing that balances answer quality, latency, and cost across smaller and frontier models.

Improve agent performance on websites without APIs, including navigation, sign-in, and checkout, with appropriate escalation when needed.

Identify and address prompt injection, data leakage, and other risks through adversarial testing and evaluations.

Advance memory, retrieval, tool selection, and decisions about when the agent should act or ask for help.

Review real conversations and turn findings into evaluation and model improvements.

What We're Looking For

At least 4 years of ML or software engineering experience at an AI lab, AI product company, or similar organization.

Hands-on experience fine-tuning or post-training LLMs and deploying the results to production.

Experience designing evaluations that teams have used to make product release decisions.

Background in at least one area: LLM red teaming, browser or computer-use agents, model routing, or production agents with tools and persistent memory.

Strong Python skills and experience with ML training and evaluation tools, plus TypeScript experience in a product codebase.

A hands-on, founding-team approach, with interest in setting technical direction and helping grow the team.

Compensation & Benefits

Salary range: $200,000 to $300,000 USD annually. Visa sponsorship is not available.

Location

On-site in San Francisco, California, United States.