# Research Software Engineer, Post Training

**Company:** [Thinking Machines Lab](https://hotfix.jobs/companies/thinking-machines-lab)
**Location:** San Francisco, CA
**Role:** ML Engineering
**Salary:** $350k – $475k/yr
**Skills:** Python, Reinforcement Learning, Containerization, Data Pipelines, Deep Learning, PyTorch, TensorFlow, JAX, Sandboxing, Machine Learning
**Posted:** 2026-09-09

> Build and operate the engineering systems that support post-training research, including reinforcement learning infrastructure, sandboxed execution, data pipelines, and agent scaffolding. The role requires strong Python and systems engineering skills, project ownership, and a relevant bachelor’s degree or equivalent experience.

## Job Description

## Responsibilities
- Embed within a research team to build, harden, and improve the systems and infrastructure required for its work.
- Design, build, and operate infrastructure including reinforcement learning training systems, sandboxing, data pipelines, and agent scaffolding.
- Lead projects end to end and contribute to large, fast-moving codebases.
- Debug systems that fail intermittently and at scale.
- Explain complex technical concepts clearly in writing.

## Requirements
- Strong proficiency in Python and strong engineering fundamentals.
- Experience leading projects end to end and writing code that others depend on.
- Bachelor's degree or equivalent experience in Computer Science, Machine Learning, Physics, Mathematics, or a related discipline.
- Strong theoretical and empirical grounding.
- Autonomous, ownership-oriented approach to advancing team goals.

## Nice-to-haves
- Experience building or iterating on sandboxed or containerized execution environments at large scale.
- Experience working in roles where team priorities determine individual priorities.
- Familiarity with at least one deep learning framework, such as PyTorch, TensorFlow, or JAX.

## Compensation and Benefits
- Expected annual salary: $350,000-$475,000 USD, depending on background, skills, and experience.
- Health, dental, and vision benefits.
- Unlimited paid time off.
- Paid parental leave.
- Relocation support.
- Visa sponsorship.

## Similar jobs

- [AI Infrastructure Engineer](https://hotfix.jobs/jobs/0d5aa4cf-a861-427b-8d8f-cba8ac94104e) - Thinking Machines Lab - San Francisco, CA - $350k – $475k/yr
- [Research, General Agents](https://hotfix.jobs/jobs/e075e229-9ae8-46ba-92a5-20a0a6d4f0db) - Thinking Machines Lab - San Francisco, CA - $350k – $475k/yr
- [Research, RL Scaling](https://hotfix.jobs/jobs/bd1b548c-7c82-4627-84a9-149150dad06d) - Thinking Machines Lab - San Francisco, CA - $350k – $475k/yr
- [Machine Learning Engineer, Multimodal Perception and Authentication](https://hotfix.jobs/jobs/26314e6e-423f-4a3f-b400-ee0b56429f64) - OpenAI - San Francisco, CA - $342k – $399k/yr
- [Software Engineer, Trainium](https://hotfix.jobs/jobs/5e1a8341-f0e9-45c8-8492-12782b38f079) - OpenAI - San Francisco, CA - $295k – $380k/yr

**Apply:** https://hotfix.jobs/jobs/168e3c8f-8577-4482-bf94-91b3d11744ba
**Canonical:** https://hotfix.jobs/jobs/168e3c8f-8577-4482-bf94-91b3d11744ba