Job at Thinking Machines Lab
Research, Tinker, RL Systems
- Employer
- Thinking Machines Lab
- Location
- San Francisco
- Workplace
- on-site
- Employment
- full-time
- Department
- Research
- Last published
- First listed on USASI
Apply on Thinking Machines Lab’s site (external site)View the posting on Thinking Machines Lab’s job board (external site)
This posting comes from Thinking Machines Lab’s own Ashby job board. USASI checks that board every day and removes this page once the employer closes the posting. The description below is Thinking Machines Lab’s text as published; USASI does not edit it, does not recruit, and does not take applications. How jobs are collected.
Job description
About Thinking Machines
The mission of Thinking Machines is to build AI that extends human will and judgment. We are training frontier models with Inkling, developing Tinker to let people make models their own, and crafting interfaces that broaden human-AI communication. We believe the future worth building is human, and we're hiring people who want to build it.
About the Role
Tinker (external site) is our fine-tuning API that empowers researchers and developers to customize frontier AI to their needs to open access to capabilities that have previously been concentrated in a handful of labs. We manage the infrastructure while allowing Tinkerers full flexibility in training models with their own data, algorithms, and for their own needs.
This role is all about building our training systems for Tinker, including RL systems, numerics, kernels, and beyond.
What You’ll Do
In this role, you'll develop frontier customization techniques and help build the best post-training engine in the industry, drawing on a whole-stack understanding recipes, data pipelines, and training systems (numerics, kernels, and beyond).
You'll engage directly with the researchers and companies pushing Tinker to its limits. This role is working with both our internal research teams as well as contributing to open science and external partners.
You’ll co-design RL algorithms and training systems across the whole stack, from RL science down to numerics and kernels, to enable anyone to post-train frontier models. You’ll debug RL runs in the wild, optimize post-training pipelines, and help users reach frontier-level results, which in turn makes our platform and models the best they can be.
Skills and Qualifications
Required qualifications:
- Bachelor’s degree or equivalent experience in Computer Science, Machine Learning, Physics, Mathematics, or a related discipline with strong theoretical and empirical grounding.
- Proficiency in Python and familiarity with deep learning frameworks (e.g., PyTorch, TensorFlow, or JAX). Comfort debugging distributed training and writing code that scales.
- Clarity in communication, an ability to explain complex technical concepts in writing.
- Strong interest in working on Tinker and increasing usefulness and adoption.
Preferred qualifications — we encourage you to apply if you meet some but not all of these:
- A strong grasp of probability, statistics, and ML fundamentals. You can look at experimental data and distinguish between real effects, noise, and bugs.
- Experience with RL training stability techniques for large runs.
- Familiarity with low-precision training and inference: numerics, quantization, and their implications for RL.
- Hands-on work with LLM serving stacks (e.g., SGLang, vLLM, TokenSpeed, or custom engines).
- Experience with scaling studies for large models.
- Contributions to open-source training or inference frameworks.
- PhD in Computer Science, Machine Learning, Physics, Mathematics, or a related discipline with strong theoretical and empirical grounding; or, equivalent industry research experience.
Logistics
- Location: This role is based in San Francisco, California.
- Compensation: Depending on background, skills and experience, the expected annual salary range for this position is $350,000 - $475,000 USD.
- Visa sponsorship: We sponsor visas. While we can't guarantee success for every candidate or role, if you're the right fit, we're committed to working through the visa process together.
- Benefits: Thinking Machines offers generous health, dental, and vision benefits, unlimited PTO, paid parental leave, and relocation support as needed.
As set forth in Thinking Machines' Equal Employment Opportunity policy, we do not discriminate on the basis of any protected group status under any applicable law.
Support Us
Help keep USASI useful.
Optional. No USASI account required. Payment takes place on the linked provider’s website (Buy Me a Coffee).
About supporting this project