← All roles

Software Engineer - Training Product

Baseten· EPD
Apply on company site
Hybrid (San Francisco), Hybrid (New York)$165k - $330kFull-timePosted 8mo ago

About the role

Software engineer role focusing on Baseten's Training product, owning features from conception to GA, working across API, UI, and infra, collaborating with research engineers and model developers to improve post-training workflows.

Responsibilities

  • Iterate like crazy
  • Design ergonomic APIs and abstractions to model complex resources and lifecycles
  • Work throughout the stack (API layer, backend and database implementation, infra layer; frontend is a plus) to implement features.
  • Fine-tune and deploy models to develop intuition around training workflows.
  • Partner closely with model developers and world-class research engineers to understand the requirements and pain points of post-training workflows.
  • Drive long-term improvements to improve reliability of systems and velocity of development
  • Fix bugs & resolve customer issues with urgency

Qualifications

  • 5+ years 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.
  • Good taste in product, particularly developer-oriented tools
  • Interest in ML/AI infrastructure and willingness to learn
  • Driven by high agency and ownership
  • Strong communication skills with the ability to bridge technical depth and business needs

Skills

Required:API designFull-stack developmentModel fine-tuning and deployment

Benefits

  • Competitive compensation, including meaningful equity
  • (U.S. only) 100% coverage of medical, dental, and vision insurance for employee and dependents
  • Flexible PTO policy including company wide Winter Break (office closed from Christmas Eve to New Year’s Day)
  • Paid parental leave
  • Fertility and family-building stipend through Carrot
  • (U.S. only) Company-facilitated 401(k)
  • Exposure to a variety of ML startups, offering unparalleled learning and networking opportunities.
Last updated Oct 21, 2026. Interested? Apply on the company site.
Baseten logo

About Baseten

Inference is everything

Effortlessly serve optimized open source & custom models on the fastest, most reliable model delivery network.

AICloud InfrastructureDeveloper ToolsInference PlatformMachine LearningModel DeploymentModel Runtimesai-infrastructure-and-development
Company profileWebsite

More open roles at Baseten