Software Engineer - Training Product
Baseten · San Francisco, CA · EPD
About this role
Baseten is hiring a mid-level Software Engineer based in San Francisco, CA. The posting calls out experience with PyTorch, Reinforcement Learning, Machine Learning, Kubernetes. Compensation is listed at $165,000–$330,000 per year.
- Role
- Software Engineer
- Function
- software engineering
- Level
- mid
- Track
- Individual contributor
- Employment
- Full-time
- Location
- San Francisco, CA
- Department
- EPD
- Posted
- Jan 22, 2026
More roles at Baseten
Job description
from Baseten careersABOUT 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 $300M Series E, backed by investors including BOND, IVP, Spark Capital, Greylock, and Conviction. Join us and help build the platform engineers turn to to ship AI products.
THE ROLE
We’re looking for a customer-obsessed software engineer to come ship with us. You’ll own features like multi-node training and products like serverless reinforcement learning (RL) from conception to MVP (and from MVP to GA!). You’ll work through the stack, architecting solutions from API and UI down to our infrastructure layer. You’ll fine tune models yourself to develop an understanding of user workflows. You’ll work closely with research engineers leveraging state-of-the-art training techniques to build experiences that accelerate model development and solve for real pain points. If you’re excited to dive deep into the training, let’s talk!
THE PRODUCT
Take a look at what we’ve built so far:
- This is an excerpt. Read the full job description on Baseten careers →