Skip to content
← Baseten

Software Engineer - Training Infrastructure

Baseten · San Francisco · New York

CHECKED 58M AGO

Infrastructureremotesenior$165k–$330k
Apply on the lab’s site ↗

At a glance

Highlights

  • Architect and lead development of training platform for frontier AI teams.
  • Design a global training scheduler for large-scale ML workloads.
  • Build reinforcement learning systems and continuous learning pipelines.

What they offer

  • Meaningful equity (U.S. only)
  • 100% coverage of medical, dental, and vision insurance for employee and dependents
  • Flexible PTO including company-wide Winter Break
  • Paid parental leave
  • Fertility and family-building stipend through Carrot (U.S. only)
  • Company-facilitated 401(k)

Ideal candidate

Infrastructure-minded builder with 5+ years experience who loves owning large-scale distributed systems end to end and partnering with researchers to turn demanding ML training needs into reliable, high-performance platforms.

Requirements

  • Bachelor's degree or higher in Computer Science or related field
  • 5+ years of experience
  • Proficiency in Go
  • Deep expertise with Kubernetes in production environments
  • Advanced understanding of distributed systems concepts and performance tuning
  • Proven experience designing observability systems
  • Experience with ML/AI workloads and MLOps platforms

Lab profile

About Baseten

Powers mission-critical AI inference to bring cutting-edge models into production.

What they build

  • Mission-critical inference platform
  • Flexible infrastructure for production models
  • Developer tooling for AI deployment

Known for

  • Applied AI research
  • Flexible infrastructure
  • Seamless developer tooling

Key facts

  • Raised Series F
More about Baseten →
Apply on the lab’s site ↗