Research, General Agents
Required skills
- Python
- PyTorch
- TensorFlow
- JAX
- distributed training
- scientific rigor
- complex codebases
- code scaling
- agentic capabilities
- tool use
Nice to have
- synthetic data pipelines
- model usability
- custom evaluations
- large-scale agentic RL infrastructure
- frontier model
- industry research experience
Role summary
Thinking Machines is seeking a Research Engineer to advance the agentic capabilities of their AI models. This role involves full development cycle ownership, building synthetic data frameworks, creating task environments for scaling training, and enhancing model usability. The ideal candidate will have strong engineering skills, proficiency in Python and deep learning frameworks, experience with distributed training, and the ability to design and interpret experiments with scientific rigor. Experience with synthetic data pipelines, agentic RL infrastructure, and technical writing is preferred. The position is based in San Francisco, CA, with a salary range of $350,000 - $475,000 USD annually, and visa sponsorship is available.
About Thinking Machines
The mission of Thinking Machines is to build AI that extends human will and judgment. We are training frontier models with Inkling, developing Tinker to let people make models their own, and crafting interfaces that broaden human-AI communication. We believe the future worth building is human, and we're hiring people who want to build it.
About the Role
This role is responsible for advancing the agentic capabilities of our models, with ownership spanning the full development cycle. Our research team is small, and the role carries a corresponding degree of autonomy and responsibility.
Note: This is an "evergreen role" that we keep open on an on-going basis to express interest in this research area. We receive many applications, and there may not always be an immediate role that aligns perfectly with your experience and skills. Still, we encourage you to apply. We continuously review applications and reach out to applicants as new opportunities open. You are welcome to reapply if you get more experience, but please avoid applying more than once every 6 months. You may also find that we put up postings for singular roles for separate, project or team specific needs. In those cases, you're welcome to apply directly in addition to an evergreen role.
What You’ll Do
Build a synthetic data framework used across the team, and create new task environments to scale training across agentic capabilities such as tool use, long-horizon tasks, and complex workflows.
Address identified gaps through data and recipe work, validating proposed changes through controlled ablations.
Enhance usability for agent capabilities end to end, translating internal and external feedback into model improvements.
Skills and Qualifications
Minimum qualifications:
Strong engineering skills, ability to contribute code and debug in complex codebases.
Ability to design, run, and interpret experiments with scientific rigor and clarity.
Proficiency in Python and familiarity with at least one deep learning framework (e.g., PyTorch, TensorFlow, or JAX). Comfortable with debugging distributed training and writing code that scales.
Bachelor’s degree or equivalent experience in Computer Science, Machine Learning, Physics, Mathematics, or a related discipline with strong theoretical and empirical grounding.
Clarity in communication, an ability to explain complex technical concepts in writing.
Preferred qualifications — we encourage you to apply even if you don’t meet all preferred qualifications, but at least some:
Experience building synthetic data pipelines and systems that were adopted by others on your team and remain in use today.
Experience owning the end-to-end cycle of identifying gaps in model usability and closing them through custom evaluations and training data.
Experience making large-scale agentic RL infrastructure reliable given the long tail of failures that surface at scale.
Experience improving the agentic capabilities of a frontier model.
PhD in Computer Science, Machine Learning, Physics, Mathematics, or a related discipline with strong theoretical and empirical grounding; or, equivalent industry research experience.
Logistics
Location: This role is based in San Francisco, California.
Compensation: Depending on background, skills and experience, the expected annual salary range for this position is $400,000 - $500,000 USD.
Visa sponsorship: We sponsor visas. 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.
Benefits: Thinking Machines offers generous health, dental, and vision benefits, unlimited PTO, paid parental leave, and relocation support as needed.
Similar roles
- Recruiting CoordinatorThinking Machines · San Francisco, CA · Onsite
- Software Engineer, Privacy & Data GovernanceThinking Machines · San Francisco, CA · Onsite
- Phase 1 Clinical Research Coordinator III (Full-time, Fri-Sun, 7PM-7AM)Arizona Liver Health · Chandler, AZ · Onsite
- 2027 Research Intern - NYCGalaxy · New York, NY · Onsite
- Clinical Research Coordinator II - Tacoma WAIterative Health · Tacoma, WA · Onsite
Senior Operations Research EngineerMIRATECH · Remote · Remote
