# Senior Product Engineer - Training Platform

**Company:** [Baseten](https://hotfix.jobs/companies/baseten)
**Location:** San Francisco, CA
**Role:** Fullstack Engineering
**Salary:** $200k – $275k/yr
**Experience:** 5+ years
**Skills:** Kubernetes, PyTorch, Deepspeed, Distributed Systems, APIs, Databases, vLLM, Fsdp, Nccl, Reinforcement Learning
**Posted:** 2026-01-22

> Senior engineer building AI training platform features like multi-node training and serverless RL. Owns full stack from API/UI to infrastructure, fine-tunes models, and collaborates with researchers. Requires 5+ years experience in software, distributed systems, and developer tools.

## Job Description

## Example Initiatives
- **Checkpointing Pipeline**: Automated checkpointing with seamless deployment to inference servers like vLLM.
- **Multinode training**: Kubernetes-based multi-node training for large models like GLM 4.7 and DeepSeek.
- **Training DX**: Enhanced metrics, CLI, and UI for fast iteration.

## Responsibilities
- Iterate rapidly on features.
- Design ergonomic APIs and abstractions.
- Work throughout the stack (API, backend, database, infra; frontend plus).
- Fine-tune and deploy models.
- Partner with model developers and research engineers.
- Drive reliability and development velocity improvements.
- Fix bugs and resolve customer issues urgently.

## Requirements
- 5+ years experience building software applications.
- Deep knowledge of web stack, databases, and distributed systems.
- Experience developing developer tooling or infrastructure products.
- Good taste in developer-oriented products.
- Interest in ML/AI infrastructure.
- High agency, ownership, and strong communication skills.

## Nice to Have
- Experience launching features through release cycles (MVP, Beta, GA).
- Model development experience (SFT, RL, LoRA, etc.).
- Familiarity with training frameworks (**PyTorch**, **DeepSpeed**, NCCL, etc.) and distributed techniques (FSDP).
- AI products/tooling experience.
- Frontend fluency.

## Benefits
- Competitive compensation with equity.
- 100% medical, dental, vision coverage.
- Generous PTO and Winter Break.
- Paid parental leave.
- 401(k).

## Similar roles

- [Senior Software Engineer](https://hotfix.jobs/jobs/7e942b76-49c3-4f2d-b3c6-7a7435b845c5) - Traba - New York, NY - $200k – $240k/yr
- [Senior Software Engineer](https://hotfix.jobs/jobs/0f69efdc-ef21-4127-a17c-3a022ecc0ff9) - FleetWorks - San Francisco, CA - $200k – $200k/yr
- [Senior Fullstack Engineer, Network Services](https://hotfix.jobs/jobs/4c9b9bd1-f710-44d1-97b5-5f0f32acc3cd) - Bluesky Social - Remote - $200k – $270k/yr
- [Senior Full Stack Engineer](https://hotfix.jobs/jobs/a7d850b6-ca5d-45f2-84c0-a74611d0fb72) - Haus - Seattle, WA - $200k – $220k/yr
- [Senior Software Engineer - Full Stack, Technical Lead](https://hotfix.jobs/jobs/159b09e1-12d8-442f-b89e-66f50c523766) - Baton - San Francisco, CA - $200k – $250k/yr

**Apply:** https://hotfix.jobs/jobs/001210d0-b5ae-42c2-b8c0-99066df19bc8
**Canonical:** https://hotfix.jobs/jobs/001210d0-b5ae-42c2-b8c0-99066df19bc8