Infrastructure Research Engineer - Distributed AI Training

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

Benefits offered by this job

Health, dental, and vision benefits

Unlimited PTO

Paid parental leave

Relocation support

Job summary

Thinking Machines in San Francisco is seeking an Infrastructure Research Engineer to enhance large-scale model training systems. Responsibilities include optimizing distributed training and advancing numerical methods to ensure model efficiency and quality. The ideal candidate will have a Bachelor's degree and experience with deep learning frameworks.

This role offers a competitive salary between $350,000 to $475,000 with generous benefits including health, dental, and unlimited PTO.

Qualifications

• Bachelor’s degree or equivalent experience in computer science, electrical engineering, or similar.
• Understanding of deep learning frameworks and their system architectures.
• Strong engineering skills to work across different stacks.

Responsibilities

• Design and optimize distributed training infrastructure for LLMs.
• Implement low-precision numerics to improve efficiency.
• Collaborate on model architectures and training recipes.

Skills

Deep learning frameworks

Collaboration

Strong engineering skills

Initiative-taking

Education

Bachelor’s degree in relevant field

Tools

PyTorch

JAX

DeepSpeed

Job description

Thinking Machines in San Francisco is seeking an Infrastructure Research Engineer to enhance large-scale model training systems. Responsibilities include optimizing distributed training and advancing numerical methods to ensure model efficiency and quality. The ideal candidate will have a Bachelor's degree and experience with deep learning frameworks.

This role offers a competitive salary between $350,000 to $475,000 with generous benefits including health, dental, and unlimited PTO.