jobs

Software Engineer - Training Product

ABOUT BASETENBaseten powers mission-critical inference for the world's most dynamic AI companies, like Cursor, Notion, OpenEvidence, Abridge, Clay, Gamma, and Writer. By uniting applied AI research, flexible infrastructure, and seamless dev

In inglese

Company

Baseten

Location

San Francisco; New York

Status

Open

Posted

2026-01-22T22:09:05.759+00:00

Baseten's training product group is hiring a software engineer to own features such as multi-node training and serverless reinforcement learning from concept through MVP to general availability. The work spans API design, backend and database implementation and the infrastructure layer, including Kubernetes scheduling and checkpointing pipelines that push models into inference servers. The engineer also fine-tunes models to understand customer workflows. Requirements include five or more years building software, knowledge of distributed systems and databases, and experience with developer tooling; familiarity with PyTorch, Megatron or FSDP is a plus.