AI Insights
OpenAI

TL, Research Inference

OpenAI · San Francisco, California, US
full-timelead (7-15 yrs)Posted 51d ago
Software EngineeringIC4IC + ManagementOn-site
StackC++PythonCUDAGPU programmingdistributed systemsinference systemskernel optimizationmemory managementprofilingbenchmarkingmulti-GPU parallelismmodel architecturelow-level debuggingbatchingscheduling

Summary

A research-enabling systems engineering lead role at OpenAI's Foundations team, focused on building high-performance inference runtimes for large-scale AI models. The role sits at the intersection of model research and systems engineering, owning distributed GPU inference infrastructure to support AI research at scale — not product serving.

About the role

About the Team

The Foundations team focuses on how model behavior changes as we scale models, data, and compute. The team studies the interactions between model architecture, optimization, and training data, and uses those insights to guide how new models are designed and trained.

About the Role

In this role, you will build the systems that enable advanced AI models to run efficiently at scale. You will operate at the intersection of model research and systems engineering, translating new architectural ideas into high-performance inference systems that surface real tradeoffs in performance, memory, and scalability.

Your work will directly influence how models are designed, evaluated, and iterated on across the research organization. By developing and evolving high-performance inference infrastructure, you will enable researchers to explore new ideas with a clear understanding of their computational and systems implications.

This is not a product-serving role. Instead, it is a research-enabling systems role focused on performance, correctness, and realism - ensuring that AI research is grounded in what can actually scale.

In this role, you will:

  • Design and build high-performance inference runtimes for large-scale AI models, with a focus on efficiency, reliability, and scalability.

  • Own and optimize core execution paths, including model execution, memory management, batching, and scheduling.

  • Develop and improve distributed inference across multiple GPUs, including parallelism strategies, communication patterns, and runtime coordination.

  • Implement and optimize inference-critical operators and kernels informed by real-world workloads.

  • Partner closely with research teams to ensure new model architectures are supported accurately and efficiently in inference systems.

  • Diagnose and resolve performance bottlenecks through profiling, benchmarking, and low-level debugging.

  • Contribute to observability, correctness, and reliability of large-scale AI systems.

You might thrive in this role if you:

  • Have experience building production inference systems, not just training or running models.

  • Are comfortable with GPU-centric performance engineering, including memory behavior and latency/throughput tradeoffs.

  • Have worked on multi-GPU or distributed systems involving batching, scheduling, or runtime coordination.

  • Can reason end-to-end about inference pipelines, from request handling through execution and output streaming.

  • Are able to understand research ideas and implement them within real system and performance constraints.

  • Enjoy solving hard, ambiguous systems problems that only emerge at scale.

  • Prefer hands-on technical ownership and execution over abstract design work.

About OpenAI

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. 

We are an equal opportunity employer, and we do not discriminate on the basis of race, religion, color, national origin, sex, sexual orientation, age, veteran status, disability, genetic information, or other applicable legally protected characteristic.

For additional information, please see OpenAI’s Affirmative Action and Equal Employment Opportunity Policy Statement.

Background checks for applicants will be administered in accordance with applicable law, and qualified applicants with arrest or conviction records will be considered for employment consistent with those laws, including the San Francisco Fair Chance Ordinance, the Los Angeles County Fair Chance Ordinance for Employers, and the California Fair Chance Act, for US-based candidates. For unincorporated Los Angeles County workers: we reasonably believe that criminal history may have a direct, adverse and negative relationship with the following job duties, potentially resulting in the withdrawal of a conditional offer of employment: protect computer hardware entrusted to you from theft, loss or damage; return all computer hardware in your possession (including the data contained therein) upon termination of employment or end of assignment; and maintain the confidentiality of proprietary, confidential, and non-public information. In addition, job duties require access to secure and protected information technology systems and related data security obligations.

To notify OpenAI that you believe this job posting is non-compliant, please submit a report through this form. No response will be provided to inquiries unrelated to job posting compliance.

We are committed to providing reasonable accommodations to applicants with disabilities, and requests can be made via this link.

OpenAI Global Applicant Privacy Policy

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.

What you'll do

1Design and build high-performance inference runtimes for large-scale AI models with a focus on efficiency, reliability, and scalability
2Own and optimize core execution paths including model execution, memory management, batching, and scheduling
3Develop and improve distributed inference across multiple GPUs, including parallelism strategies, communication patterns, and runtime coordination
4Implement and optimize inference-critical operators and kernels informed by real-world workloads
5Partner closely with research teams to ensure new model architectures are supported accurately and efficiently in inference systems
6Diagnose and resolve performance bottlenecks through profiling, benchmarking, and low-level debugging
7Contribute to observability, correctness, and reliability of large-scale AI systems

Requirements

Proven experience building and owning production-grade inference systems for large-scale AI models, not just training or evaluation pipelines
Deep expertise in GPU-centric performance engineering including memory hierarchy, latency/throughput tradeoffs, and kernel-level optimization
Hands-on experience with multi-GPU or distributed inference systems, including parallelism strategies, runtime coordination, and communication patterns
Ability to reason end-to-end about inference pipelines from request handling through execution and output streaming
Strong cross-functional collaboration skills to partner with AI researchers and translate new model architectures into performant, correct inference systems

Nice to have

vLLM
TensorRT
Triton
NCCL
FlashAttention
PyTorch
JAX
MPI
RDMA/InfiniBand networking
observability tooling

Role overview

Role family
Software Engineering
Level
IC4 — ml_ai
Experience
7–15 years
Type
Hybrid (IC + Management)
Remote policy
On-site
Visa sponsorship
Not offered

Tech stack analysis

LANGUAGES
C++PythonCUDA
FRAMEWORKS
PyTorchJAXTriton
INFRASTRUCTURE
Multi-GPU clustersNCCLInfiniBand/RDMANVIDIA A100/H100 GPUs
TOOLS
profilers (Nsight, perf)benchmarking toolslow-level debuggersobservability/monitoring tooling

Salary estimate

$320K – $550K
AI-estimated salary range
Confidence72%
Reasoning

OpenAI is known for exceptionally high compensation packages. A TL (Tech Lead) role at the intersection of systems engineering and AI research in San Francisco at a frontier AI lab commands top-of-market pay. Based on known OpenAI compensation bands (Levels.fyi, industry data), senior/staff-level systems engineers at OpenAI earn $300K–$450K+ in total comp; a TL designation with research infrastructure ownership likely places this at $320K–$550K+ total compensation including base, bonus, and equity (RSUs/profit participation units). Base salary alone is likely $200K–$280K.

See the AI-estimated salary range for this role

Sign up free →

Green flags

5 items
Direct impact on frontier AI research — infrastructure built here influences how next-generation models are designed and evaluated across the entire research orggrowth

Discover all 5 green flags for this role

Sign up free →

Benefits breakdown

See all benefits organized by category — health, financial, time off & more

Sign up free →

Hiring insights

JD quality
8/10
Urgency
medium
Autonomy
high
Team size
small (2-5)

See JD quality score, hiring urgency & team details

Sign up free →

Red flags

PRO4 items
No salary range disclosed, which is legally required in California (Labor Code §432.3) — a notable omission for a San Francisco-based rolecompensation

See all 4 red flags — what the JD isn't telling you

Sign up free →

Interview insights

PRO
Rounds
6
Duration
6 wks
Difficulty
very hard
Take-home
No

Get full interview breakdown — rounds, likely topics & prep tips

Sign up free →

Career path

PRO
Next roles
Principal Engineer, AI InfrastructureDirector of Research EngineeringStaff Research Scientist (Systems)

See where this role leads — full career progression

Sign up free →
About the company

OpenAI is the AI research laboratory behind GPT-4, ChatGPT, DALL-E, and the Codex API. With over 200 million weekly active ChatGPT users, OpenAI is at the forefront of large language model development and deployment. The company pursues a mission of building safe artificial general intelligence that benefits all of humanity.

HQSan Francisco, CA, USA
Interview difficultyvery hard
Build vs Maintainboth
Cross-functionalYes