Senior ML Training Infrastructure Engineer

Baseten

San Francisco (CA)

On-site

USD 165,000 - 330,000

Full time

14 days+

Get more replies from employers

Send a job-specific resume in minutes.

Benefits offered by this job

Competitive compensation, including meaningful equity
100% coverage of medical, dental, and vision insurance
Flexible PTO policy and paid parental leave
Company-facilitated 401(k)

Job summary

Baseten is looking for a Software Engineer to join their Training Infrastructure team in San Francisco. You'll architect and lead the development of the training platform, enabling developers to deploy and monitor workloads efficiently. The ideal candidate has a Bachelor's degree, strong proficiency in Go, extensive experience with Kubernetes, and knowledge of ML workload management. Benefits include competitive compensation, medical insurance coverage, and a flexible PTO policy.

Qualifications

  • Bachelor's degree or higher in Computer Science or related field.
  • Proficiency in Go, with Python experience a plus.
  • Deep expertise with Kubernetes in production environments.
  • Extensive experience with major cloud providers (AWS, GCP).
  • Advanced understanding of distributed systems concepts.
  • Proven experience designing observability systems.
  • Experience with ML/AI workloads and MLOps platforms highly valued.

Responsibilities

  • Design and architect scalable infrastructure systems for ML training platform.
  • Partner closely with developers and research engineers.
  • Drive long‑term improvements for reliability and velocity of development.
  • Lead technical discussions and mentor junior engineers.

Skills

Proficiency in Go
Experience with Python
Deep expertise with Kubernetes
Experience with major cloud providers (AWS, GCP)
Advanced understanding of distributed systems
Experience with ML/AI workloads

Education

Bachelor's degree or higher in Computer Science or related field

Job description

Baseten is looking for a Software Engineer to join their Training Infrastructure team in San Francisco. You'll architect and lead the development of the training platform, enabling developers to deploy and monitor workloads efficiently. The ideal candidate has a Bachelor's degree, strong proficiency in Go, extensive experience with Kubernetes, and knowledge of ML workload management. Benefits include competitive compensation, medical insurance coverage, and a flexible PTO policy.
Get your free, confidential resume review.
or drag and drop your file here.
Similar jobs

Similar jobs worth comparing

Software Engineer - Training Infrastructure
Software Engineer - Training Infrastructure

The Consensus • New York (NY)

On-site
USD 110,000 - 150,000
Competitive compensation, including equity
100% coverage of medical, dental, and vision insurance
Flexible PTO policy including company-wide Winter Break
+4
Software Engineer - Training Infrastructure
Software Engineer - Training Infrastructure

Baseten • San Francisco (CA)

On-site
USD 165,000 - 330,000
Competitive compensation, including meaningful equity
100% coverage of medical, dental, and vision insurance
Flexible PTO policy and paid parental leave
+1
Software Engineer, ML Training Platform (Remote)
Software Engineer, ML Training Platform (Remote)

Baseten • United States

Remote
USD 180,000 - 230,000
Remote-first culture
In-person team summits three times a <
ML Training Infrastructure Engineer — Equity
ML Training Infrastructure Engineer — Equity

Baseten • United States

Remote
USD 180,000 - 240,000
Equity
Full medical/dental/vision coverage
Senior ML Systems Engineer - Training Infra
Senior ML Systems Engineer - Training Infra

OpenAI • California (MO)

On-site
USD 295,000 - 380,000
Relocation assistance
Senior ML Training Systems Engineer - Distributed GPU Infra
Senior ML Training Systems Engineer - Distributed GPU Infra

Baseten • San Francisco (CA)

On-site
USD 150,000 - 200,000
Competitive compensation, including equity
100% coverage of medical, dental, and vision insurance
Generous PTO policy
+2
ML Training Infrastructure Engineer
ML Training Infrastructure Engineer

The Consensus • New York (NY)

On-site
USD 110,000 - 150,000
Competitive compensation, including equity
100% coverage of medical, dental, and vision insurance
Flexible PTO policy including company-wide Winter Break
+4
Senior ML Systems Engineer: Training Infra
Senior ML Systems Engineer: Training Infra

Neura Market • San Francisco (CA)

On-site
USD 295,000 - 380,000
Relocation assistance
Senior ML Infrastructure Engineer - Scalable GPU Training
Senior ML Infrastructure Engineer - Scalable GPU Training

ATOMS Careers page • San Francisco (CA)

On-site
USD 224,000 - 280,000
Medical, Dental, Vision Insurance
401(k)
Unlimited Flexible Time Off
ML Infra Engineer: Scale Training & Inference (Hybrid)
ML Infra Engineer: Scale Training & Inference (Hybrid)

Lattice, Inc. • San Francisco (CA)

Hybrid
USD 200,000 - 280,000
Competitive salary
Premium health, dental, and vision insurance
Unlimited PTO
+2