Thinking Machines Lab
Research Software Engineer, Post Training
$350K–$475K
San Francisco · On-site · Full-time · Posted Sep 9, 2026
- Education
- Bachelor’s degree or equivalent experience in Computer Science, Machine Learning, Physics, Mathematics, or a related discipline.
- Visa sponsorship
- Visa sponsorship is available; success is not guaranteed.
About the job
Build and operate the engineering systems and infrastructure that post-training research teams rely on.
What you'll do
- Embed in a research team to build, harden, and improve its systems and infrastructure.
- Design, build, and operate infrastructure, including RL training systems, sandboxing, data pipelines, and agent scaffolding.
What they ask for
- Bachelor’s degree or equivalent experience in Computer Science, Machine Learning, Physics, Mathematics, or a related discipline, with strong theoretical and empirical grounding.
- Experience leading projects end to end, working in large fast-moving codebases, and writing code others have depended on and built on.
- Strong proficiency in Python and engineering fundamentals, including debugging systems that fail intermittently and at scale.
- Clear communication and ability to explain complex technical concepts in writing; autonomous drive and an ownership mindset.
Nice to have
- Experience building or iterating with sandboxed or containerized execution environments at a large scale.
- Familiarity with at least one deep learning framework, such as PyTorch, TensorFlow, or JAX.
Pay
$350,000 - $475,000 USD (base salary)
Tools
Python, PyTorch, TensorFlow, JAX
AI focus
Build infrastructure for post-training research, including RL training systems, data pipelines, and agent scaffolding.
Model-written summary · the original posting is authoritative