Research, Tinker, RL Systems
Thinking Machines Lab9 hours ago
Base Salary
$350k - $475k/yr
Responsibilities
- Develop frontier customization techniques and build Tinker’s post-training engine.
- Co-design reinforcement-learning algorithms and training systems across RL science, numerics, and kernels.
- Debug RL runs, optimize post-training pipelines, and improve training stability and scalability.
- Work with internal researchers, companies, external partners, and Tinker users to achieve frontier-level results.
- Contribute to open science and help increase the usefulness and adoption of Tinker.
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 PyTorch, TensorFlow, or JAX.
- Ability to debug distributed training and write code that scales.
- 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 ML 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
- Health, dental, and vision benefits.
- Unlimited paid time off.
- Paid parental leave.
- Relocation support as needed.
- The role is based in San Francisco, California.
- The company sponsors visas and is committed to working through the visa process with qualified candidates, though success is not guaranteed.
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.