Thinking Machines Lab

Thinking Machines Lab

Research, Post-Training

San Francisco

H1B sponsorship available$350k-$475kDetected 78 days ago
PythonExpressMachine LearningDeep LearningTensorFlowPyTorchStatisticsA/B TestingLogisticsResearchCommunication

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.

Responsibilities

  • 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.
  • 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.
  • Develop and tune the recipe: iterate on post-training recipes, consisting of a collection of datasets, training stages, and hyperparameters.

Requirements

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

Nice to have

  • 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.
  • 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.
  • or, equivalent industry research experience.

Skills

  • Thinking Machines Lab's mission is to empower humanity through advancing collaborative general intelligence.

Compensation

  • Depending on background, skills and experience, the expected annual salary range for this position is $350,000 - $475,000 USD.

Benefits

  • Thinking Machines offers generous health, dental, and vision benefits, unlimited PTO, paid parental leave, and relocation support as needed.
  • Proficiency in Python and familiarity with at least one deep learning framework (e.g., PyTorch, TensorFlow, or JAX).
  • Bachelor's degree or equivalent experience in Computer Science, Machine Learning, Physics, Mathematics, or a related discipline with strong theoretical and empirical grounding.
  • Prior experience with RLHF, RLAIF, preference modeling, or reward learning for large models.

Visa & Work Authorization

  • 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.
  • We sponsor visas.

This listing is sourced directly from Thinking Machines Lab's careers page and normalized into a canonical job model.