Thinking Machines Lab

Research, Tinker, RL Systems

Thinking Machines Lab
Apply
3 hours ago

Base Salary

$350k - $475k/yr

Responsibilities

  • Develop frontier model customization techniques and build Tinker’s post-training engine.
  • Co-design RL algorithms and training systems across RL science, numerics, kernels, and related layers.
  • Debug RL runs, optimize post-training pipelines, and improve training stability for large-scale runs.
  • Work with internal research teams, external partners, researchers, and companies using Tinker.
  • Help users achieve frontier-level results and contribute to open science.

Requirements

  • 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 such as PyTorch, TensorFlow, or JAX.
  • Experience debugging distributed training and writing scalable code.
  • Clear written communication and the ability to explain complex technical concepts.
  • Strong interest in working on Tinker and increasing its usefulness and adoption.
  • Preferred qualifications include probability, statistics, and ML fundamentals; RL training stability; low-precision training and inference; LLM serving stacks; large-model scaling studies; open-source training or inference contributions; or a PhD or equivalent industry research experience.

Benefits

  • Generous health, dental, and vision benefits.
  • Unlimited PTO and paid parental leave.
  • Relocation support as needed.
  • Based in San Francisco, California.
  • Annual base salary range of $350,000-$475,000 USD.
  • Visa sponsorship is available, with a commitment to work through the visa process for the right fit.

Tech Stack

PythonPyTorchTensorFlow

Categories

AI ResearchML Engineering
Thinking Machines Lab

About Thinking Machines Lab

201-500 employees

Thinking Machines Lab develops AI and generative AI software and conducts applied research to help organizations make data-driven decisions. The company builds products and data science solutions for enterprise use cases, pairing foundational models with practical tooling and services across industries. It is privately held and headquartered in San Francisco.

Contact me