ML Infrastructure Engineer

AI Infrastructure EngineerThe mission of Thinking Machines is to build AI that extends human will and judgment. We are training frontier models with Inkling, developing Tinker to let people make models their own, and crafting interfaces that broaden human-AI communication. We believe the future worth building is human, and we're hiring people who want to build it.We're hiring an AI Infrastructure Engineer to keep our post-training and reinforcement learning (RL) systems fast, reliable, and easy for researchers to iterate on. Think of this as a production engineering or site reliability role built around model training: you'll own the health of the training runs, clusters, and pipelines that power post-training and RL at Thinking Machines.You'll work side by side with research teams during active model runs — debugging failures in real time, hardening infrastructure against the next class of problem, and building the tooling and automation that let researchers spend their time on the science instead of babysitting jobs. This role has real ownership: you'll be the person a research team calls when a run stalls at 2am, and the person who makes sure it doesn't happen again.What You'll DoOwn the reliability, performance, and uptime of large-scale post-training and RL training jobs, from launch through completionPartner directly with research teams during active model runs, embedding with them to unblock training and speed up iterationDebug failures across the full stack — accelerators, networking, storage, schedulers, and training frameworks — and drive issues to root causeBuild monitoring, alerting, and automated recovery so runs self-heal or fail fast instead of silently stallingImprove checkpointing, fault tolerance, and job scheduling so hardware failures cost minutes, not days of computeBuild internal tools that reduce toil and improve cluster utilization across post-training and RL workloadsParticipate in an on-call rotation supporting production model runsWrite postmortems and turn recurring failure patterns into permanent infrastructure fixesMinimum Qualifications4+ years of experience as a production engineer, site reliability engineer, or infrastructure engineer operating large-scale distributed systems in productionTrack record debugging complex failures across distributed systems — networking, hardware, kernel, or scheduler issuesStrong software engineering skills in Python and/or Go/C++, with the judgment to know when to script a fix versus build a systemSolid grounding in Linux systems internals and networking fundamentalsComfortable owning production systems, including participating in on-call rotationsPreferred QualificationsExperience operating GPU or TPU training clusters at scaleFamiliarity with post-training and RL techniques (e.g., RLHF, PPO, DPO) and the infrastructure challenges specific to them, such as reward model serving, rollout generation, and mixed training/inference workloadsExperience with distributed training frameworks (e.g., PyTorch, Ray) and job schedulers (e.g., Slurm, Kubernetes)Experience with high-performance networking (e.g., InfiniBand, RDMA, NCCL) and its role in distributed training performanceExperience building observability tooling purpose-built for ML training, not just general infrastructureA track record of thriving in fast-changing, research-driven environments where priorities shift with the scienceLogisticsLocation: This role is based in San Francisco, CA.Compensation: Depending on background, skills and experience, the expected annual salary range for this position is $350,000 - $475,000 USD.Visa sponsorship: We sponsor visas. While we can't guarantee success for every candidate or role, if you're the right fit, we're committed to working through the visa process together.Benefits: Thinking Machines offers generous health, dental, and vision benefits, unlimited PTO, paid parental leave, and relocation support as needed.