Thinking Machines Lab's mission is to empower humanity through advancing collaborative general intelligence. They are seeking an infrastructure research engineer to design and build core systems for scalable training of large models using reinforcement learning.
Responsibilities:
- Design, build, and optimize the infrastructure that powers large-scale reinforcement learning and post-training workloads
- Improve the reliability and scalability of RL training pipeline, distributed RL workloads, and training throughput
- Develop shared monitoring and observability tools to ensure high uptime, debuggability, and reproducibility for RL systems
- Collaborate with researchers to translate algorithmic ideas into production-grade training pipelines
- Build evaluation and benchmarking infrastructure that measures model progress on helpfulness, safety, and factuality
- Publish and share learnings through internal documentation, open-source libraries, or technical reports that advance the field of scalable AI infrastructure