Infra Research Engineer: Distributed AI Training Systems

Thinking Machines Lab; San Francisco (CA); On-site; USD 350,000 - 475,000; Full time

Benefits offered by this job

Generous health benefits

Unlimited PTO

Paid parental leave

Relocation support

Job summary

Thinking Machines Lab, located in San Francisco, is seeking an Infrastructure Research Engineer to design and optimize distributed training systems that enable efficient training of large models. The ideal candidate will possess strong engineering skills and a background in using deep learning frameworks.

This position offers an annual salary range of $350,000 to $475,000 USD, along with generous health benefits and unlimited PTO. Join us in shaping the future of AI!

Qualifications

• Bachelor’s degree or equivalent experience in computer science or related field.
• Strong engineering skills and ability to debug in complex codebases.
• Understanding of deep learning frameworks and system architectures.

Responsibilities

• Design and optimize distributed training systems for large-scale workloads.
• Develop high-performance optimizations to maximize efficiency.
• Collaborate with researchers and engineers to build scalable infrastructure.

Skills

Engineering skills

Understanding of deep learning frameworks

Collaboration in cross-functional teams

Initiative and action-oriented mindset

Education

Bachelor’s degree in relevant field

Tools

PyTorch

JAX

Job description

Thinking Machines Lab, located in San Francisco, is seeking an Infrastructure Research Engineer to design and optimize distributed training systems that enable efficient training of large models. The ideal candidate will possess strong engineering skills and a background in using deep learning frameworks.

This position offers an annual salary range of $350,000 to $475,000 USD, along with generous health benefits and unlimited PTO. Join us in shaping the future of AI!