AI Engineer: Build Agentic AI Features for Production

US Health Partners, LLC; San Francisco (CA); Hybrid; USD 150,000 - 210,000; Full time

Benefits offered by this job

Medical, dental, vision insurance for

Job summary

Baseten is seeking an AI Engineer to join the Training Product team, building AI-driven features for customers training and post-training models on our platform. You will work with research engineers to scope and convert ideas into customer-facing products, providing autonomy and ownership over execution, guardrails, and performance improvements.

You will help Baseten evolve internal workflows into agentic solutions, shipping features that increase velocity across engineering, research, and GTM

Qualifications

• 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.
• Comfort working autonomously in a fast-moving environment with limited structure.
• Ability to move between customer-facing product work and internal tooling and automation.
• Strong communication skills, with the ability to bridge technical depth and business needs.

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 so you know whether a change actually improved output quality.
• Work throughout the stack (API layer, backend, agent orchestration, frontend) to implement features end to end.
• Use Baseten's own training and inference products yourself to develop intuition around customer workflows.
• Identify where AI can replace manual process across the company and build the thing rather than write the proposal.
• Fix bugs and resolve customer issues with urgency.

Skills

AI/LLM-powered products

Python

Strong software engineering

Autonomy in fast-moving env

Communication skills

LangChain (tooling)

Tools

LangChain

Job description

Baseten is seeking an AI Engineer to join the Training Product team, building AI-driven features for customers training and post-training models on our platform. You will work with research engineers to scope and convert ideas into customer-facing products, providing autonomy and ownership over execution, guardrails, and performance improvements.

You will help Baseten evolve internal workflows into agentic solutions, shipping features that increase velocity across engineering, research, and GTM