Research Software Engineer, Post Training

Job summary

Responsibilities

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.

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

What they're looking 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 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.

Tech stack

• Python
• PyTorch
• TensorFlow
• JAX