368,530open jobs
9,432companies
50,439added this week
Browse all
Salary
$220k – $260k per year
Location
Remote/Hybrid (San Francisco, United States)
Seniority
Senior · 5+ years exp
Employment
Full-Time
Overview
Company
Impact
Profile match
Baseten is an AI infrastructure platform designed to help developers and machine learning teams deploy, serve, and scale open-source and custom AI models. The platform provides performant, low-latency inference infrastructure alongside developer tools like Truss, an open-source model packaging framework. Headquartered in San Francisco, California, Baseten enables companies to run state-of-the-art models in production seamlessly without managing underlying cloud infrastructure.

ABOUT BASETEN

Baseten powers mission-critical inference for the world's most 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. We're growing quickly and recently raised our $1.5B Series F, led by Altimeter Capital, Conviction Partners, and Spark Capital. Join us and help build the platform engineers turn to to ship AI products.

THE ROLE:

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 you'll raise the ceiling on how Baseten itself uses AI internally, turning manual workflows into agentic ones that make every other team faster.

You'll work directly with our research engineers to scope and build products, taking ideas from a research loop that already works internally to something customers can run themselves. This is a hands-on role with real autonomy. You'll pick the problems worth solving, build the harnesses, execution flows, and guardrails that make AI systems reliable, and own the results. If you've been shipping agents and want that to be the job, let's talk.

EXAMPLE INITIATIVES:

Take a look at these blog posts written by members of our team:

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.

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 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.

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.

BENEFITS

  • Competitive compensation, including meaningful equity.

  • 100% coverage of medical, dental, and vision insurance for employee and dependents

  • Flexible PTO policy including company wide Winter Break (our offices are closed from Christmas Eve to New Year's Day!)

  • Paid parental leave

  • Fertility and family-building stipend through Carrot

  • Company-facilitated 401(k)

  • Exposure to a variety of ML startups, offering unparalleled learning and networking opportunities.

Apply now to embark on a rewarding journey in shaping the future of AI! If you are a motivated individual with a passion for machine learning and a desire to be part of a collaborative and forward-thinking team, we would love to hear from you.

At Baseten, we are committed to fostering a diverse and inclusive workplace. We provide equal employment opportunities to all employees and applicants without regard to race, color, religion, gender, sexual orientation, gender identity or expression, national origin, age, genetic information, disability, or veteran status.

We are an Equal Opportunity Employer and will consider qualified applicants with criminal histories in a manner consistent with applicable law (by example, the requirements of the San Francisco Fair Chance Ordinance, where applicable).

Free account
Stop reading job ads. Get the ones that fit.
One free account turns this page into a shortlist built around your stack, your level and your pay.
Match on every job. Stack, seniority, pay and location, scored against your profile.
368,530 open roles. Read straight off company career pages, refreshed every day.
Unlimited applications. Every one you send is tracked in one place, on-site or on a company board.
3 tailored CVs a month. Rewritten for the exact job you are applying to. Included free.
Create a free account
Free forever. No card. Under a minute.

Your match

How well do you fit this role?
Two answers are enough for a real match. No account needed.
Check my fit
Answers stay in this browser until you create an account.

Recommended for you based on this role

Similar stack
Same company
San Francisco
$27k – $60k per year (Estimated) • In office • Moscow
Python
Python
pySpark
Databases
ClickHouse
AI/ML
CatBoost
LightGBM
MLFlow
PyTorch
Spark
DevOps
Amazon S3
Analytics
ETL/ELT
Apply
$7k – $21k per year • Remote • Contractor • Yekaterinburg
Python
AI/ML
ChatGPT
Claude
LLM
Midjourney
Management
n8n
Zapier
Apply
$51k – $111k per year (Estimated) • In office • Full-Time • 3+ years exp • Bachelor's Degree • Cork
Bash
Node JS
PowerShell
Python
JavaScript
AI/ML
AI Agents
DevOps
Rest API
Windows Server
Marketing
Salesforce
Apply
System Engineer 9 hours ago
$10k – $40k per year (Estimated) • In office • Full-Time • 2+ years exp • Bengaluru
Java
Python
Management
Jira
Apply
Copilot AI Engineer 9 hours ago
$22k – $62k per year (Estimated) • In office • Full-Time • 3+ years exp • Bachelor's Degree • Hyderabad
C#
Python
TypeScript
AI/ML
A2A
AI Agents
Copilot
Function Calling
Google AI Studio
Model Context Protocol
OpenAI
RAG
Semantic Search
Semantic Search
DevOps
Azure
Rest API
Analytics
Power BI
Management
Power Apps
Power Automate
Apply
$165k – $330k per year • Remote/Hybrid • Full-Time • San Francisco • Toronto • New York • Montreal
AI/ML
Cursor
DeepSeek
LLM
SGLang
TensorRT
TensorRT-LLM
vLLM
Management
Notion
Apply
$190k – $240k per year • Remote/Hybrid • Full-Time • San Francisco • New York
JavaScript
TypeScript
AI/ML
Cursor
Frontend
Next.js
React Three Fiber
React.js
Three.JS
Design
Figma
Apply
Product Designer 5 days ago
$225k – $290k per year • Remote/Hybrid • Full-Time • San Francisco • New York
JavaScript
TypeScript
AI/ML
Cursor
Frontend
React.js
DevOps
Git
Apply
$170k – $220k per year • Remote/Hybrid • Full-Time • 8+ years exp • San Francisco • New York
Go
AI/ML
Cursor
LLM
Post-training
Apply
$200k – $400k per year • Remote/Hybrid • Full-Time • 1+ year exp • San Francisco • New York
AI/ML
Cursor
JAX
LLM
PyTorch
Ray
SGLang
TensorRT
TensorRT-LLM
vLLM
InfiniBand
Post-training
SFT
DevOps
Kubernetes
SLURM
Chips/EDA
PoC Library
Apply
$223k – $424k per year (Estimated) • In office • Bachelor's Degree • San Francisco
AI/ML
AI Agents
LLM
Recommender Systems
Apply
$160k – $283k per year • Equity • In office • 5+ years exp • San Francisco
AI/ML
AI Agents
Apply
$83k – $188k per year (Estimated) • In office • 2+ years exp • San Francisco
Python
AI/ML
AI Agents
LLM Guardrails
Model Context Protocol
DevOps
Terraform
Cybersecurity
Crowdstrike
GDPR
Least Privilege
Okta
SentinelOne
Management
Google Workspace
Slack
Apply
$171k – $273k per year • In office • Full-Time • 8+ years exp • PhD • San Francisco • Washington
AI/ML
A2A
Agentforce
AI Agents
Model Context Protocol
DevOps
AWS
GCP
Marketing
Salesforce
Apply
Security GRC Analyst 2 hours ago
$119k – $268k per year (Estimated) • Remote/Hybrid • 4+ years exp • Bachelor's Degree • San Francisco
AI/ML
Ignite
PyTorch
Cybersecurity
ISO 27001
NIST CSF
SOC 2
Apply
See all jobs
This is one of many
368,530 more open roles from verified company boards, updated every day.