Research, Tinker, RL Systems

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

Benefits offered by this job

Health, dental, vision benefits

Unlimited PTO

Paid parental leave

Relocation support

Job summary

Thinking Machines Lab Inc. in San Francisco is seeking a highly skilled ML researcher to build and optimize the Tinker training systems, including RL components, numerics, and kernels. You will design scalable training pipelines and contribute to frontier AI capabilities.

Collaborate with internal research teams and external partners, co-design RL algorithms, and help deliver post-training tooling that makes our platform perform at the frontier.

Qualifications

• Bachelor’s degree or equivalent in CS/ML/Physics/Math with strong theoretical and empirical grounding.
• Proficiency in Python and familiarity with deep learning frameworks (PyTorch, TensorFlow, or JAX).
• Clear written communication and ability to explain complex concepts.
• Strong interest in working on Tinker and expanding usefulness and adoption.

Responsibilities

• Develop frontier customization techniques and build training systems (RL, numerics, kernels).
• Engage directly with researchers and external partners pushing Tinker to its limits.
• Co-design RL algorithms and training systems across the whole stack to enable post-training frontier models.
• Debug RL runs, optimize post-training pipelines, and help users reach frontier-level results.

Skills

Python

DL frameworks

Communication

Tinker interest

Education

Bachelor’s degree or equivalent in CS/ML/Physics/Math

PhD in CS/ML/Physics/Math (preferred)

Tools

RL serving stacks

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

Tinker 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.