AI Engineer

About Baseten

Baseten powers mission-critical inference for dynamic AI companies like Cursor, Notion, OpenEvidence, Abridge, Clay, Gamma, and Writer. By uniting applied AI research, flexible infrastructure, and seamless developer tooling, we enable companies operating at the frontier of AI to bring cutting-edge models into production.

The Role

We're looking for an AI Engineer to join our Training Product team. You'll build AI-driven product features for customers training and post-training frontier models on our platform, and raise the ceiling on how Baseten uses AI internally, turning manual workflows into agentic ones that accelerate every team.

You'll work directly with our research engineers to scope and build products, taking ideas from working internal research loops to features customers can run themselves.

Responsibilities

• Build and ship agentic product experiences, including chat-style and assistant-like interfaces, from prototype to GA.
• Design the harnesses, execution flows, and guardrails that make AI systems reliable in production.
• Build internal automation and AI tooling that measurably increases the velocity of engineering, research, and go-to-market teams.
• Partner with research engineers to scope product opportunities out of internal research workflows and turn them into customer-facing features.
• Define and instrument evals to track output quality improvements.
• Work throughout the stack (API layer, backend, agent orchestration, frontend) to implement features end-to-end.
• Use Baseten's own training and inference products to build intuition around customer workflows.
• Identify where AI can replace manual processes across the company and build solutions directly.
• Fix bugs and resolve customer issues with urgency.

Requirements

• 5+ years of experience building and shipping software applications.
• Demonstrated experience building AI or LLM-powered products, agents, or agentic workflows that real users depend on.
• Strong software engineering fundamentals.
• Ability to build accurate mental models of how systems work under the hood, including models and harnesses.
• Proficiency in Python, with fluency in at least one other language.
• Comfort working autonomously in a fast-moving environment with limited structure.
• Ability to move between customer-facing product work and internal tooling/automation.
• Strong communication skills, bridging technical depth and business needs.

Nice to Have

• Experience as a founding engineer or early employee at a startup.
• Experience building evals, agent observability, or tooling for non-deterministic systems.
• Familiarity with agent frameworks and harnesses (LangChain, Claude Code, Codex, OpenCode, MCP).
• Experience with model development methods like supervised fine-tuning, reinforcement learning, synthetic data generation, LoRA, and full fine-tunes.
• Frontend fluency.