AI Engineer

Job summary

Responsibilities

Are you the person on your team who builds the agent everyone else ends up using? We're looking for an AI Engineer to join our Training Product team and do that at Baseten. You'll build AI-driven product features for the customers training and post-training frontier models on our platform, and…

• 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 so you know whether a change actually improved output quality.

What they're looking for

• 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 and the ability to clear a real technical bar, not just prompt well.
• Ability to build accurate mental models of how systems work under the hood, including the models and harnesses you're building on.
• Proficiency in Python, with fluency in at least one other language.

Tech stack

• Spark
• LangChain
• Python