Research Software Engineer, Post Training
Thinking Machines Lab2 months ago
Base Salary
$350k - $475k/yr
Responsibilities
- Embed within a research team to build, harden, and improve the systems and infrastructure required for its success.
- Design, build, and operate RL training systems, sandboxing, data pipelines, and agent scaffolding.
- Lead projects end to end and write code that other engineers and researchers depend on.
- Debug systems that fail intermittently and at scale.
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 others depend on.
- Strong proficiency in Python and strong engineering fundamentals.
- Clear written communication and the ability to explain complex technical concepts.
- Preferred: experience with sandboxed or containerized execution environments at large scale.
- Preferred: familiarity with at least one deep learning framework such as PyTorch, TensorFlow, or JAX.
Benefits
- Annual salary range of $350,000-$475,000 USD.
- Visa sponsorship is offered, with commitment to work through the visa process for suitable candidates.
- Generous health, dental, and vision benefits.
- Unlimited PTO and paid parental leave.
- Relocation support as needed.
- Role is based in San Francisco, California.
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.