Software Engineer - Training Product

Remote San Francisco FullTime

Posted 7mo ago

Remote Work Policy

Fully remote

Employment Type

FullTime

Categories

Applied AI Engineer

About the job

Baseten is seeking a customer-obsessed software engineer to join their team and contribute to the development of mission-critical AI inference platforms. In this role, you will own features from conception to launch, working across the entire technology stack from API and UI down to the infrastructure layer. You will have the opportunity to fine-tune models, gain a deep understanding of user workflows, and collaborate closely with research engineers to build cutting-edge experiences that accelerate model development and address real-world pain points. If you are excited about diving deep into AI model training and building impactful products, this is the role for you.

Responsibilities

  • Design ergonomic APIs and abstractions for complex resources and lifecycles.
  • Implement features across the full stack, including API, backend, database, and infrastructure.
  • Fine-tune and deploy models to understand training workflows.
  • Collaborate with model developers and research engineers to address post-training workflow challenges.
  • Drive long-term improvements in system reliability and development velocity.
  • Resolve customer issues and fix bugs with urgency.

Requirements

  • 5+ years of experience building software applications.
  • Deep knowledge of web stack, databases, and distributed systems.
  • Experience developing developer tooling or infrastructure products.
  • Good product sense, especially for developer-oriented tools.
  • Interest in ML/AI infrastructure and a willingness to learn.
  • High agency and ownership.
  • Strong communication skills to bridge technical depth and business needs.

Benefits

  • Competitive compensation and meaningful equity.
  • 100% coverage of medical, dental, and vision insurance for employee and dependents.
  • Flexible PTO policy with company-wide Winter Break.
  • Paid parental leave.
  • Fertility and family-building stipend.
  • Company-facilitated 401(k).
  • Exposure to a variety of ML startups for learning and networking.

About Baseten

Get new AI jobs in your inbox

A weekly digest of the newest AI engineering roles.

© 2026 AI Job Board. All rights reserved.