Post-Training Research Engineer
Baseten · San Francisco
About this role
**Post-Training Research Engineer** **About Baseten** Baseten powers mission-critical inference for leading AI companies like Cursor, Notion, OpenEvidence, Abridge, Clay, Gamma, and Writer. Recently raised $1.5B Series F led by Altimeter Capital, Conviction Partners, and Spark Capital. **The Role** Join the Post-Training team to build in-house tooling supporting custom model training. Work across the full stack—from Kubernetes and storage systems to PyTorch distributed tensor computation and GPU kernels. Focus on training diverse model architectures with various techniques efficiently at scale. **What We're Looking For** • Deep understanding of modern ML techniques for training transformers • Advanced experience with PyTorch, TensorFlow, JAX, or similar • Detailed knowledge of transformer training parallelism strategies (data, tensor, pipeline, context) • Ability to profile and optimize distributed GPU programs • Roofline analysis capability on transformer training setups • Experience with HPC platforms (Slurm, Ray, Kubernetes, Dask) • Familiarity with cluster networking (Infiniband, RoCE, GPUDirect) • Strong OS fundamentals (processes, files, containerization, networking) • Creativity and willingness to challenge assumptions **Benefits** • Competitive compensation with meaningful equity • 100% medical, dental, vision coverage (U.S.) • Flexible PTO + company-wide Winter Break • Paid parental leave • Fertility/family-building stipend • 401(k) facilitation (U.S.) • Exposure to diverse ML startup ecosystem **Equal Opportunity Employer** — We welcome qualified applicants from all backgrounds.
Listing freshness
CronJobs last confirmed this listing 2h ago. If its source stops confirming the opening for seven days, this page is removed from active inventory.