Research, Tinker, RL Systems
Thinking Machines Lab20 days ago
Base Salary
$350k - $475k/yr
Responsibilities
- Develop frontier 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 and performance.
- Work directly with internal research teams, external partners, researchers, and companies using Tinker.
- Contribute to open science and help users achieve frontier-level model results.
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.
- Ability to debug distributed training and write 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: strong probability, statistics, and machine learning fundamentals; RL training stability experience; low-precision training and inference experience; familiarity with LLM serving stacks; large-model scaling studies; open-source training or inference contributions; or a relevant PhD or equivalent industry research experience.
Benefits
- San Francisco, California-based role.
- Annual salary range of $350,000-$475,000 USD.
- Visa sponsorship is offered, with a commitment to work through the visa process for the right fit, though success is not guaranteed for every candidate or role.
- Generous health, dental, and vision benefits.
- Unlimited PTO, paid parental leave, and relocation support as needed.
Categories
AI ResearchML Engineering
About Thinking Machines Lab
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.