Baseten

Software Engineer - Training Product

Baseten
Apply
7 months ago
San Francisco, CA, USA or New York, NY, USASenior
H1B Sponsor

Base Salary

$165k - $330k/yr

Responsibilities

  • Own features such as multi-node training and serverless reinforcement learning from conception through MVP and GA.
  • Design ergonomic APIs and abstractions for complex resources and lifecycles.
  • Implement features across the API, backend, database, infrastructure, and optionally frontend layers.
  • Fine-tune and deploy models to understand training workflows.
  • Partner with model developers and research engineers on post-training requirements and pain points.
  • Drive improvements to system reliability and development velocity.
  • Fix bugs and resolve customer issues urgently.

Requirements

  • 5+ years of experience building software applications.
  • Deep knowledge of the web stack, databases, and distributed systems.
  • Experience developing developer tooling or infrastructure products for external or internal users.
  • Strong product judgment, particularly for developer-oriented tools.
  • Interest in ML/AI infrastructure and willingness to learn.
  • High agency and ownership.
  • Strong communication skills and ability to bridge technical depth with business needs.
  • Experience launching features and products through MVP, Beta, and GA release cycles is preferred.
  • Experience with supervised fine-tuning, reinforcement learning, synthetic data generation, LoRA, or full fine-tunes is preferred.
  • Familiarity with open-source training frameworks and distributed training techniques is preferred.
  • Experience developing AI products, tooling, or agents is preferred.
  • Frontend fluency is preferred.

Benefits

  • Competitive compensation including meaningful equity, with no specific base salary stated.
  • 100% medical, dental, and vision insurance coverage for employees and dependents.
  • Flexible PTO and company-wide Winter Break from Christmas Eve through 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 and learning and networking opportunities.
Baseten

About Baseten

201-500 employees

Inference is everything. Baseten is an AI infrastructure platform giving you the tooling, expertise, and hardware needed to bring great AI products to market - fast. Our proprietary Inference Stack utilizes the cutting-edge of performance research combined with highly performant and reliable infrastructure to give you out-of-the-box global availability with 99.99% of uptime.