Tech Lead Manager - Model Training

ABOUT BASETENBaseten provides the infrastructure, tooling, and expertise needed to bring great AI products to market - fast. Backed by top investors including IVP, Spark Capital, Greylock, and Conviction, we’re trusted by leading AI-driven innovators like Writer, Abridge, Bland, Patreon, Descript, Retool, and Zed to deliver industry-leading performance, security, and reliability for their mission-critical workloads. With our recent $75M Series C funding, we’re growing fast to make AI accessible across all products. THE ROLEAs a Tech Lead Manager of the Training team at Baseten, you’ll lead a team of engineers building the core systems that power large-scale training and fine-tuning of foundation models. Your team will be responsible for designing scalable, reliable, and efficient infrastructure - covering distributed training frameworks, GPU scheduling, and training pipelines—enabling both Baseten and our customers to train and adapt models at scale. You’ll balance hands-on technical contributions with people management, setting the technical direction while fostering the growth and success of your team. You’ll also play a key role in defining Baseten’s platform roadmap by identifying common infrastructure needs and turning them into reusable, self-serve capabilities. RESPONSIBILITIESLead, mentor, and grow a team of engineers building Baseten’s training infrastructureDefine and drive the technical strategy for large-scale training systems, with a focus on scalability, reliability, and efficiencyArchitect and optimize distributed training pipelines across heterogeneous GPU/accelerator environmentsBalance hands-on contributions (system design, code reviews, prototyping) with people leadership and career developmentEstablish best practices for training workflows, distributed systems design, and high-performance model evaluationCollaborate with Product and Platform Engineering to translate customer and internal needs into reusable infrastructure and APIsDevelop processes that ensure consistent, reliable, and on-time delivery of high-quality systemsStay ahead of the curve on advancements in training efficiency (FSDP, ZeRO, parameter-efficient training, hardware-aware scheduling) and bring them into productionREQUIREMENTSBachelor’s degree in Computer Science, Engineering, or related field, or equivalent experience5+ years of experience in ML infrastructure, distributed systems, or ML platform engineering, including 2+ years in a tech lead or manager roleStrong expertise in distributed training frameworks and orchestration (FSDP, DDP, ZeRO, Ray, Kubernetes, Slurm, or similar)Hands-on experience building or scaling training infrastructure for LLMs or other foundation modelsDeep understanding of GPU/accelerator hardware utilization, mixed precision training, and scaling efficiencyProven ability to lead and mentor technical teams while delivering complex infrastructure projectsExcellent communication skills, with the ability to bridge technical depth and business needsNICE TO HAVEExperience with multi-tenant, production-grade ML platformsFamiliarity with cluster management, GPU scheduling, or elastic resource scalingKnowledge of advanced model adaptation techniques (LoRA, QLoRA, RLHF, DPO)Contributions to open-source distributed training or ML infrastructure projectsExperience building developer-friendly APIs or SDKs for ML workflowsCloud-native infrastructure experience (AWS, GCP, Azure, containerization, orchestration)BENEFITSCompetitive compensation package. This is a unique opportunity to be part of a rapidly growing startup in one of the most exciting engineering fields of our era. An inclusive and supportive work culture that fosters learning and growth. Exposure to a variety of ML startups, offering unparalleled learning and networking opportunities. Apply now to embark on a rewarding journey in shaping the future of AI! If you are a motivated individual with a passion for machine learning and a desire to be part of a collaborative and forward-thinking team, we would love to hear from you. At Baseten, we are committed to fostering a diverse and inclusive workplace. We provide equal employment opportunities to all employees and applicants without regard to race, color, religion, gender, sexual orientation, gender identity or expression, national origin, age, genetic information, disability, or veteran status. Apply to this Job