Research, Post-Training

Posted 18 Days Ago
Be an Early Applicant
San Francisco, CA, USA
Hybrid
350K-475K Annually
Entry level
Artificial Intelligence • Information Technology
The Role
Conduct post-training research for frontier AI models by developing and tuning training recipes, designing meaningful evaluations, debugging distributed training, scaling methodologies, and exploring novel datasets. Analyze experimental results, distinguish signal from noise, and publish research, code, datasets, and insights. The role combines theoretical research with high-performance engineering and focuses on alignment, human-AI collaboration, and model usefulness and safety.
Summary Generated by Built In
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

The role of post-training researchers sits at the core of our roadmap. This is the critical bridge between raw model intelligence and a system that is actually useful, safe, and collaborative for humans.

This role blends fundamental research and practical engineering, as we do not distinguish between the two roles internally. You will be expected to write high-performance code and read technical reports. It’s an excellent fit for someone who enjoys both deep theoretical exploration and hands-on experimentation, and who wants to shape the foundations of how AI learns.

Note: This is an "evergreen role" that we keep open on an on-going basis to express interest in this research area. We receive many applications, and there may not always be an immediate role that aligns perfectly with your experience and skills. Still, we encourage you to apply. We continuously review applications and reach out to applicants as new opportunities open. You are welcome to reapply if you get more experience, but please avoid applying more than once every 6 months. You may also find that we put up postings for singular roles for separate, project or team specific needs. In those cases, you're welcome to apply directly in addition to an evergreen role.

What You’ll Do
  • Develop and tune the recipe: iterate on post-training recipes, consisting of a collection of datasets, training stages, and hyperparameters. Measure how recipe choices affect various metrics.

  • Iterate on evals: post-training involves a never-ending loop of defining a set of evaluations, optimizing them, and then realizing your existing evals don’t capture what matters. You’ll be responsible for both making numbers go up, and making sure the numbers are meaningful.

  • Debug and understand: while tuning the details of a training configuration, we often observe results that don’t quite make sense. You’ll be responsible for both getting things to work, and developing a deeper understanding, which we can bring to the next problem.

  • Scale and explore: post-training will involve a combination of scaling the existing methodologies and developing new ones. We’ll want to both measure how performance metrics scale with dataset size, and explore using a completely different kind of training dataset.

  • Publish and present research that moves the entire community forward. Share code, datasets, and insights that accelerate progress across industry and academia.

Skills and Qualifications

Minimum qualifications:

  • Proficiency in Python and familiarity with at least one deep learning framework (e.g., PyTorch, TensorFlow, or JAX). Comfortable with debugging distributed training and writing code that scales.

  • Bachelor’s degree or equivalent experience in Computer Science, Machine Learning, Physics, Mathematics, or a related discipline with strong theoretical and empirical grounding.

  • Clarity in communication, an ability to explain complex technical concepts in writing.

Preferred qualifications — we encourage you to apply even if you don’t meet all preferred qualifications, but at least some:

  • A strong grasp of probability, statistics, and ML fundamentals. You can look at experimental data and distinguish between real effects, noise, and bugs.

  • Prior experience with RLHF, RLAIF, preference modeling, or reward learning for large models.

  • Experience managing or analyzing human data collection campaigns or large-scale annotation workflows.

  • Research or engineering contributions in alignment, data-centric AI, or human-AI collaboration.

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

Skills Required

  • Proficiency in Python
  • Familiarity with at least one deep learning framework, such as PyTorch, TensorFlow, or JAX
  • Experience debugging distributed training and writing scalable code
  • Bachelor's degree or equivalent experience in Computer Science, Machine Learning, Physics, Mathematics, or a related discipline
  • Strong theoretical and empirical grounding
  • Ability to communicate clearly and explain complex technical concepts in writing
  • Strong grasp of probability, statistics, and machine learning fundamentals
  • Experience with RLHF, RLAIF, preference modeling, or reward learning for large models
  • Experience managing or analyzing human data collection campaigns or large-scale annotation workflows
  • Research or engineering contributions in alignment, data-centric AI, or human-AI collaboration
  • PhD in Computer Science, Machine Learning, Physics, Mathematics, or a related discipline, or equivalent industry research experience
Am I A Good Fit?
beta
Get Personalized Job Insights.
Our AI-powered fit analysis compares your resume with a job listing so you know if your skills & experience align.

The Company
HQ: Singapore
91 Employees

What We Do

Thinking Machines Lab is an artificial intelligence research and product company. We're building a future where everyone has access to the knowledge and tools to make AI work for their unique needs and goals. While AI capabilities have advanced dramatically, key gaps remain. The scientific community's understanding of frontier AI systems lags behind rapidly advancing capabilities. Knowledge of how these systems are trained is concentrated within the top research labs, limiting both the public discourse on AI and people's abilities to use AI effectively. And, despite their potential, these systems remain difficult for people to customize to their specific needs and values. To bridge the gaps, we're building Thinking Machines Lab to make AI systems more widely understood, customizable and generally capable. We are scientists, engineers, and builders who've created some of the most widely used AI products, including ChatGPT and Character.ai, open-weights models like Mistral, as well as popular open source projects like PyTorch, OpenAI Gym, Fairseq, and Segment Anything.

Similar Jobs

CHAI AI Logo CHAI AI

Senior Research Engineer, Post-Training

Artificial Intelligence • Software • Generative AI
In-Office
Palo Alto, CA, United States
20 Employees
609K-900K Annually
In-Office
San Francisco, CA, USA
2217 Employees
250K-500K Annually

Lightning AI Logo Lightning AI

Senior Research Engineer, LLM Training & Post-Training

Artificial Intelligence • Machine Learning • Software
Remote or Hybrid
3 Locations
50 Employees
165K-310K Annually

Thinking Machines Lab Logo Thinking Machines Lab

Software Engineer

Artificial Intelligence • Information Technology
Hybrid
San Francisco, CA, USA
91 Employees
350K-475K Annually

Similar Companies Hiring

Onshore Thumbnail
Artificial Intelligence • Fintech • Software • Financial Services
New York, New York
60 Employees
Blee Thumbnail
Artificial Intelligence • Marketing Tech • Software
New York, New York
30 Employees
Vega Thumbnail
Artificial Intelligence • Automotive • Insurance • Transportation
US
43 Employees

Sign up now Access later

Create Free Account

Please log in or sign up to report this job.

Create Free Account