← All roles

Research Engineer, Infrastructure, Training Systems

Thinking Machines Lab· Research Infrastructure (ML Infra)
Apply on company site
San Francisco, CA$350k - $475kfull-timePosted 11mo ago

About the role

Infrastructure research engineer role focused on designing and building core distributed training systems to enable scalable, efficient training of large models. End-to-end ownership of the training stack to speed experimentation and scientific progress, with emphasis on performance, reliability, and collaboration with research teams.

Responsibilities

  • 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.

Qualifications

  • Bachelor’s degree or equivalent experience in computer science, electrical engineering, statistics, machine learning, physics, robotics, or similar.
  • Strong engineering skills, ability to contribute performant, maintainable code and debug in complex codebases.
  • Understanding of deep learning frameworks (e.g., PyTorch, JAX) and their underlying system architectures.
  • Thrive in a highly collaborative environment involving many cross-functional partners and subject matter experts.
  • Bias for action with a mindset to initiative to work across different stacks and teams to ship features.

Skills

Required:Distributed systems designHigh-performance optimizationFrameworks knowledge (PyTorch, JAX)Collaboration across teams

Benefits

  • generous health, dental, and vision benefits
  • unlimited PTO
  • paid parental leave
  • relocation support as needed
Last updated Jul 8, 2026. Interested? Apply on the company site.
Thinking Machines Lab logo

About Thinking Machines Lab

Connectionism: Research Blog by Thinking Machines Lab

Thinking Machines Lab is an artificial intelligence research and product company. We’re building a future where everyone has access to the knowledge and tools to make AI work for their unique needs and goals.

AIartificial intelligencebessemer-ai-infrastructure-2026deep learningliquid-ai-recruiting-targetmachine learningproductreinforcement-learning-platforms
Company profileWebsite

More open roles at Thinking Machines Lab