Software Engineer, ML Training Platform (Remote)

Baseten; United States; Remote; USD 180,000 - 230,000; Full time

Benefits offered by this job

Remote-first culture

In-person team summits three times a <

Job summary

Baseten is seeking a software engineer to ship developer-focused ML infrastructure. You will design and implement features across API, backend, and infra, enabling multi-node training, checkpointing, and seamless deployment of checkpoints to inference servers.

You will collaborate with research engineers to apply state-of-the-art training techniques, iterate rapidly, and deliver user-friendly tooling. You’ll work from a remote-friendly setup, with opportunities to join in-person summits and

Qualifications

• 5+ years of building software applications
• Experience building developer tooling or infrastructure products
• Excellent written and verbal communication
• Deep knowledge of web stack, databases, and distributed systems
• Interest in ML/AI infrastructure and willingness to learn

Responsibilities

• Own features from conception to MVP and GA
• Architect solutions across API, backend, database, infra, and frontend
• Fine-tune models to understand user workflows
• Collaborate with research engineers on training techniques
• Improve training DX and pipeline reliability
• Fix bugs and respond to customer issues with urgency

Skills

5+ years experience

Developer tooling/infrastructure

Strong communication

Web stack & distributed systems

ML/AI infrastructure interest

Product sense for dev tools

Model training methods (Fine-Tuning,RL

Open source training stacks

Distributed training (DeepSpeed)

Frontend fluency

Tools

NCCL

PyTorch

Megatron

DeepSpeed

HF Trainer

Job description

Baseten is seeking a software engineer to ship developer-focused ML infrastructure. You will design and implement features across API, backend, and infra, enabling multi-node training, checkpointing, and seamless deployment of checkpoints to inference servers.

You will collaborate with research engineers to apply state-of-the-art training techniques, iterate rapidly, and deliver user-friendly tooling. You’ll work from a remote-friendly setup, with opportunities to join in-person summits and