Software Engineer, ML Systems & Training Architecture

OpenAI Inc.
San Francisco, United States of America
yesterday

Role details

Contract type
Permanent contract
Employment type
Full-time (> 32 hours)
Working hours
Regular working hours
Languages
English
Compensation
$ 295K

Job location

San Francisco, United States of America

Tech stack

Clean Code Principles
Artificial Intelligence
Software Quality
Code Review
Software Debugging
Distributed Systems
Machine Learning
Graphics Processing Unit (GPU)
Codebase
Machine Learning Operations

Job description

As a Senior Software Engineer, ML Systems & Training Infrastructure, you will be a deeply hands-on engineering force multiplier for the robotics team. You will help keep the training framework and surrounding infrastructure healthy, review and improve code quickly, debug failures across ML systems and infrastructure, and unblock researchers and engineers when the path from idea to working training job gets rough.

We're looking for people who love writing, reading, reviewing, and fixing code; who can get productive quickly in unfamiliar systems; and who bring strong practical judgment without a lot of ego or process overhead.

This role will be based in San Francisco, CA and be expected in office 5 days per week and offer relocation assistance to new employees.

In this role, you will:

  • Review, improve, and clean up code across training frameworks and adjacent infrastructure.
  • Identify risky or low-quality changes before they land, and raise the code quality bar without slowing the team down.
  • Debug issues across ML training systems, GPUs, clusters, networking, and related infrastructure.
  • Help researchers and engineers unblock broken training jobs, flaky workflows, and brittle internal tooling.
  • Improve the reliability, maintainability, and usability of the robotics team's training framework.
  • Move quickly on practical engineering problems that directly affect team velocity.

Requirements

  • Have strong software engineering fundamentals and excellent code review judgment.
  • Have experience with ML systems, training frameworks, GPUs, distributed systems, infrastructure, or similarly complex technical environments.
  • Read and debug unfamiliar codebases quickly, and enjoy getting to root cause.
  • Ship high-quality code with strong velocity and pragmatic judgment.
  • Are low-ego, responsive, and motivated by helping researchers and engineers move faster.
  • Prefer being a highly effective hands-on IC over driving broad process-heavy initiatives.
  • Have experience reviewing messy, fast-moving, or AI-generated codebases.

About the company

The OpenAI Robotics team is focused on unlocking general-purpose robotics and pushing towards AGI-level intelligence in dynamic, real-world settings. Working across the entire model stack, we integrate cutting-edge hardware and software to explore a broad range of robotic form factors. We strive to seamlessly blend high-level AI capabilities with the constraints of physical systems to improve peoples' lives., OpenAI is an AI research and deployment company dedicated to ensuring that general-purpose artificial intelligence benefits all of humanity. We push the boundaries of the capabilities of AI systems and seek to safely deploy them to the world through our products. AI is an extremely powerful tool that must be created with safety and human needs at its core, and to achieve our mission, we must encompass and value the many different perspectives, voices, and experiences that form the full spectrum of humanity., At OpenAI, we believe artificial intelligence has the potential to help people solve immense global challenges, and we want the upside of AI to be widely shared. Join us in shaping the future of technology.

Apply for this position