Research Software Engineer, Post Training at Thinking Machines Lab | CA, US | Rezi

Research Software Engineer, Post Training at Thinking Machines Lab

Research Software Engineer, Post Training

Thinking Machines Lab · CA, US

1 weeks ago

Research Software Engineer, Post Training

Thinking Machines Lab · CA, US

14 days ago
Resume preview

Impress employers and recruiters.
Choose from hundreds of resume examples.

Target Resume Now
Resume preview

Tailor your resume to this Research Software Engineer, Post Training role.

Rezi rewrites your resume against Thinking Machines Lab's job description. Free.

Resume score gauge reading 58 out of 100

Don't guess if your resume is good enough.

See how it scores against the Research Software Engineer, Post Training posting at Thinking Machines Lab — free, in seconds.

About the Role

This role is responsible for building and strengthening the engineering foundations that our post-training research teams depend on. You'll embed within a research team, and build or improve the systems required for the team to succeed. Our research teams are small, and the role carries a corresponding degree of autonomy and responsibility.

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 research teams depend on, including RL training systems, sandboxing, data pipelines, and agent scaffolding.

Requirements

  • Experience leading projects end to end, working in large fast-moving codebases, and writing code others have depended and built on.
  • Strong proficiency in Python and strong engineering fundamentals, with experience debugging systems that fail intermittently and at scale.
  • Clarity in communication, an ability to explain complex technical concepts in writing.
  • Strong autonomous drive to progress towards the team’s goals with an ownership mindset.
  • 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, and a track record of success in such a role.
  • Familiarity with at least one deep learning framework (e.g., PyTorch, TensorFlow, or JAX).

Skills

  • Python
  • Deep learning frameworks (e.g., PyTorch, TensorFlow, or JAX)

Location

  • San Francisco, California

Work Type

  • Onsite

Experience Level

  • Mid-level

Education Level

  • Bachelor’s degree or equivalent experience in Computer Science, Machine Learning, Physics, Mathematics, or a related discipline with strong theoretical and empirical grounding.

Salary/Compensations

  • 350000-475000 USD

Benefits

  • Generous health, dental, and vision benefits
  • Unlimited PTO
  • Paid parental leave
  • Relocation support as needed
  • Visa sponsorship

About the Company

  • 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.

Equal Opportunity

  • As set forth in Thinking Machines' Equal Employment Opportunity policy, we do not discriminate on the basis of any protected group status under any applicable law.