Research Engineer, Infrastructure, Kernels

About Thinking Machines Lab

Thinking Machines Lab empowers humanity by advancing collaborative general intelligence.

What you'd do

• Design custom ML kernels for LLM operations
• Reduce memory bandwidth bottlenecks
• Collaborate with research teams on optimizations
• Maintain a library of reusable kernels and benchmarks

What they're looking for

• Bachelor’s degree in a relevant field
• Strong engineering skills
• Proficiency in CUDA, CuTe, Triton
• Understanding of deep learning frameworks

Perks

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