Your browser does not support javascript! Please enable it, otherwise web will not work for you.

RL Systems - Infra

Home > Python programming jobs

RL Systems - Infra in USA

  • Thinking Machines Lab
  • Full time
  • Email
  • San Francisco, CA

Responsibilities

  • Design, build, and optimize infrastructure for large-scale reinforcement learning and post-training workloads.
  • Improve the reliability, scalability, and throughput of distributed RL training pipelines and workloads.
  • Develop monitoring and observability tools supporting uptime, debuggability, and reproducibility.
  • Collaborate with researchers to turn algorithmic ideas into production-grade training pipelines.
  • Build evaluation and benchmarking infrastructure for model helpfulness, safety, and factuality.
  • Share technical learnings through documentation, open-source libraries, or technical reports.

Requirements

  • Bachelor’s degree or equivalent experience in computer science, electrical engineering, statistics, machine learning, physics, robotics, or a similar field.
  • Strong engineering skills, including the ability to write performant, maintainable code and debug complex codebases.
  • Understanding of deep learning frameworks and their underlying system architectures, including PyTorch or JAX.
  • Experience with large-scale language models, reinforcement learning workloads, high-performance or reliability engineering, distributed training frameworks, cluster orchestration, monitoring, observability, or large-scale ML research and infrastructure is preferred.
  • Experience with methods such as PPO, DPO, RLHF, or reward modeling is preferred.

Benefits

  • The role is based in San Francisco, California.
  • Annual base salary is $425,000–$475,000 USD.
  • Visa sponsorship is available for qualified candidates, subject to the company’s stated visa-process caveat.
  • Benefits include health, dental, and vision coverage, unlimited PTO, paid parental leave, and relocation support as needed.
  • This is an evergreen role reviewed on an ongoing basis, and applicants are asked not to reapply more than once every six months.

Salary: $425k - $475k/yr

Thinking Machines Lab

Thinking Machines Lab pioneers advanced AI research, blending cutting‑edge theory with practical applications to solve complex problems across industries. Our interdisciplinary team pushes the boundaries of machine learning, neural architecture, and cognitive modeling to deliver innovative soluti...

Similar positions

Full-Stack Engineer - Platform Engineering

  • ElevenLabs
  • Full time
  • Remote
  • 10/15/2026
  • Worldwide

Embedded Systems Test Engineer

  • Geotab
  • Full time
  • Others
  • 10/15/2026
  • Kraków, Poland

Infrastructure Engineer

  • ElevenLabs
  • Full time
  • Remote
  • 10/15/2026
  • Europe, UK

Senior Software Engineer - Data Engineer

  • Tekion
  • Full time
  • Others
  • 10/15/2026
  • Salary: Competitive
  • Bengaluru, India

Safety Engineer (Back-End Leaning)

  • ElevenLabs
  • Full time
  • Remote
  • 10/15/2026
  • UK, Poland