Research Engineer, Infrastructure, Training Systems

Design, implement, and optimize distributed training systems that scale across thousands of GPUs and nodes for large-scale training workloads.

Develop high-performance optimizations to maximize throughput and efficiency.

Develop reusable frameworks and libraries to improve training reproducibility, reliability, and scalability for new model architectures.

Establish standards for reliability, maintainability, and security, ensuring systems are robust under rapid iteration.

Collaborate with researchers and engineers to build scalable infrastructure.

Publish and share learnings through internal documentation, open-source libraries, or technical reports that advance the field of scalable AI infrastructure.

Location: This role is based in San Francisco, California.

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.