O

OpenAI

Builds AI models and the ChatGPT assistant

Software Engineer - Accelerators

Applied Machine LearningConfirmed live in the last 24 hours
Full-timeSan Francisco, CA, USA$295k - $380k /yrIn-person

About the Job

Requirements

  • At least 3 years of experience working on AI infrastructure, including kernels, systems, or hardware-software co-design.
  • Hands-on experience with data-center-scale AI accelerator platforms, such as TPUs, custom silicon, or exploratory architectures.
  • Strong understanding of kernels, sharding, runtime systems, or distributed scaling techniques.
  • Familiarity with optimizing large language models, convolutional neural networks, or recommender models for hardware efficiency.
  • Experience with performance modeling, system debugging, and software stack adaptation for novel architectures.
  • Ability to work across multiple levels of the stack, rapidly prototype solutions, and navigate ambiguity during early hardware bring-up phases.
  • Interest in exploring alternatives to mainstream accelerators and shaping future AI compute.

Responsibilities

  • Prototype and enable the AI software stack on new, exploratory accelerator platforms.
  • Optimize large-scale model performance, including large language models, recommender systems, and distributed AI workloads, across diverse hardware environments.
  • Develop kernels, sharding mechanisms, and system scaling strategies tailored to emerging accelerators.
  • Collaborate on model-code-level optimizations, such as PyTorch, and lower-level optimizations to improve performance on non-traditional hardware.
  • Perform system-level performance modeling, debug bottlenecks, and drive end-to-end optimization.
  • Work with hardware teams and vendors to evaluate alternatives to existing platforms and adapt the software stack to their architectures.
  • Contribute to runtime improvements, compute/communication overlap, and scaling efforts for frontier AI workloads.

Desired Qualifications

  • Exposure to mobile accelerators is welcome; experience enabling data-center-scale AI hardware is preferred.

About the Company

OpenAI

Builds AI models and the ChatGPT assistant

OpenAI researches and builds artificial intelligence models and turns them into products for people, developers, and businesses, with a stated mission of making sure artificial general intelligence benefits all of humanity. Its launch of ChatGPT in 2022 brought generative AI into everyday use. People use ChatGPT to ask questions, write, code, and analyze files, developers build OpenAI's models into their own apps through its API, and its Codex agent works through coding tasks on its own. OpenAI earns money from ChatGPT subscriptions, business plans, and API usage. Sam Altman, Elon Musk, and others co-founded it in 2015; it is based in San Francisco and overseen by a nonprofit, the OpenAI Foundation.

Company size
1,001-5,000
Company stage
Late Stage VC
Total funding
$207.5B
Headquarters
San Francisco, CA
Founded
2015

Recently Posted Jobs

Explore more opportunities at OpenAI

See more jobs