Research Engineer, Infrastructure, Numerics

About Thinking Machines Lab

Thinking Machines Lab empowers humanity through collaborative general intelligence development.

What you'd do

• Design and optimize training infrastructure for large-scale LLMs
• Implement low-precision numerics to enhance efficiency
• Develop kernels and communication primitives
• Collaborate with teams on model architectures and training recipes

What they're looking for

• Bachelor’s degree in a relevant field
• Understanding of deep learning frameworks (e.g., PyTorch, JX
• Strong engineering skills for maintainable code and complex

Perks

• Generous health, dental, and vision benefits
• Unlimited PTO
• Paid parental leave