Research Software Engineer, Post Training
About The Position
Thinking Machines is building AI that extends human will and judgment, including frontier models with Inkling, Tinker for customizing models, and interfaces for broader human-AI communication. The Research Software Engineer, Post Training builds and strengthens the engineering foundations that post-training research teams depend on. The engineer embeds within a research team and builds or improves the systems the team needs to succeed. Research teams are small, so the role carries a corresponding degree of autonomy and responsibility. The position is based in San Francisco, California. The expected annual salary range is $350,000–$475,000 USD, depending on background, skills, and experience. Thinking Machines sponsors visas and offers generous health, dental, and vision benefits, unlimited PTO, paid parental leave, and relocation support as needed.
Requirements
- 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 that others have depended on and built upon.
- Strong proficiency in Python and strong engineering fundamentals.
- Experience debugging systems that fail intermittently and at scale.
- Clear communication skills and the ability to explain complex technical concepts in writing.
- Autonomous drive to progress toward team goals with an ownership mindset.
Nice To Haves
- Experience building or iterating with sandboxed or containerized execution environments at a large scale.
- Experience in a role where the team’s priorities set yours, with a track record of success in such a role.
- Familiarity with at least one deep learning framework, such as PyTorch, TensorFlow, or JAX.
Responsibilities
- Embed within a research team to build, harden, and improve the systems and infrastructure required for the team to succeed.
- Design, build, and operate infrastructure used by research teams, including RL training systems, sandboxing, data pipelines, and agent scaffolding.
Benefits
- Health benefits
- Dental benefits
- Vision benefits
- Unlimited PTO
- Paid parental leave
- Relocation support as needed
Stand Out From the Crowd
Upload your resume and get instant feedback on how well it matches this job.
What This Job Offers
Job Type
Full-time
