Logo for Rockset

Software Engineer, Kernel Performance & AI Tooling

Role overview

Qualifications

  • Strong systems or tooling engineering experience, with a background in low-level software, performance optimization, or infrastructure.
  • Experience with developer tooling, debugging infrastructure, profiling, observability, or workflow design for technical users.
  • Depth in kernel development, accelerator architecture, compiler systems, or related performance-critical domains.
  • Familiarity with AI-assisted systems, agentic workflows, post-training, or reinforcement learning for engineering or research applications.

Responsibilities

  • Build developer tooling and workflows that make kernel development and performance optimization faster, more scalable, and easier to debug, integrate, and deploy.
  • Develop observability, diagnostics, and validation infrastructure that makes AI-assisted optimization systems more interpretable, reliable, and effective.
  • Optimize production kernels end to end by formulating optimization problems, running search loops, analyzing bottlenecks, debugging generated implementations, and landing improvements into production.
  • Design abstractions, interfaces, and automation systems that accelerate kernel optimization, correctness validation, and hardware-software co-design.

About the company

Rockset logo

Rockset

Computer Software / SaaS

Our Vision We believe that a data-driven world has the potential to make life better for everyone. Enterprises are still struggling to use complex data primarily because real-world data is messy and cannot be put to use easily. We are bridging the gap by changing the way data is stored, processed and accessed for making better, faster data-driven decisions and data powered apps. Empowering enterprises to unleash all their data is a difficult challenge that inspires us every day. Our Team Rockset's team has deep expertise in storage, data management and distributed systems. Members of our team started the Hadoop File System project back in 2006 that helped ignite the big data movement. We previously founded and led the creation of Facebook's online social graph serving engine and graph search projects - TAO, and Unicorn - that power all of Facebook's user facing and search products. Our team also helped build the original backend for Gmail at Google. On the enterprise side, members of our team have experience launching VMware's vSAN and building the industry's first nested virtualization in the cloud at Ravello. We intimately understand data, cloud and scale as well as the challenges and opportunities it creates for enterprises.

Company details

Company typeScaleup
IndustryComputer Software / SaaS
Company size51 - 200

Your match analysis

See how your profile stacks up against this role.

We compared the job requirements to your profile to show where you're strong and where you fall short.

Job description

About the Team

OpenAI’s Hardware organization develops silicon and system-level solutions designed for the unique demands of advanced AI workloads. The team is responsible for building the next generation of AI-native silicon while working closely with software and research partners to co-design hardware tightly integrated with AI models. In addition to delivering production-grade silicon for OpenAI’s supercomputing infrastructure, the team also creates custom design tools and methodologies that accelerate innovation and enable hardware optimized specifically for AI.

About the Role

We are looking for a systems-minded engineer to help advance our kernel development, performance engineering, and hardware-software co-design capabilities, with a particular focus on AI-assisted workflows and tooling. This person will work at the intersection of kernel optimization, developer tooling, observability, and research infrastructure, helping us improve both how production kernels are built and optimized, and how future hardware-software systems are designed and evaluated. The role is ideal for someone who is excited by low-level performance work, but also sees AI and automation as powerful tools for accelerating engineering velocity. You will help define the future of kernel engineering in the era of AI-assisted development.

In this role, you may:

  • Build developer tooling and workflows that make kernel development and performance optimization faster, more scalable, and easier to debug, integrate, and deploy.

  • Develop observability, diagnostics, and validation infrastructure that makes AI-assisted optimization systems more interpretable, reliable, and effective.

  • Optimize production kernels end to end by formulating optimization problems, running search loops, analyzing bottlenecks, debugging generated implementations, and landing improvements into production.

  • Design abstractions, interfaces, and automation systems that accelerate kernel optimization, correctness validation, and hardware-software co-design.

  • Improve AI-assisted optimization systems for specialized tasks through better datasets, evaluations, benchmarking, and research infrastructure.

  • Partner across research and engineering teams to turn new ideas into practical systems spanning production needs and long-term infrastructure strategy.

You might thrive in this role if you have:

  • Strong systems or tooling engineering experience, with a background in low-level software, performance optimization, or infrastructure.

  • Experience with developer tooling, debugging infrastructure, profiling, observability, or workflow design for technical users.

  • Depth in kernel development, accelerator architecture, compiler systems, or related performance-critical domains.

  • Familiarity with AI-assisted systems, agentic workflows, post-training, or reinforcement learning for engineering or research applications.

  • Strong experimental judgment, comfort with ambiguity, and the ability to move fluidly between research exploration and production execution.

  • Interest in compilers, DSLs, program synthesis, or AI for systems.

Preferred profile

The ideal candidate is a strong systems and tooling engineer with real depth in kernels and accelerators. They are comfortable working across software and hardware boundaries, can reason deeply about performance, abstractions, and system design, and have hands-on experience optimizing code for GPUs, high-performance CPUs, or custom accelerators. They view AI not as the end product, but as a force multiplier for engineering productivity and system optimization.

To comply with U.S. export control laws and regulations, candidates for this role may need to meet certain legal status requirements as provided in those laws and regulations.

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.

Apply once. Then go straight to the hiring manager.

After you apply, unlock the direct contact details of the people who actually make the call. A quick follow-up makes you 5x more likely to land an interview.

MR

Marcus Rivera

Chief Revenue Officer

m.rivera@company.com
linkedin.com/in/marcusrivera
Unlocked after you apply
·

Software Engineer Related jobs

Other jobs at Rockset

Premium

Reach out to the hiring manager directly.

Gain access to the contact details of the hiring managers who actually decide, and reach out to network with them directly. That, plus more when you upgrade:

  • Full match report with fit score and gaps
  • Career diagnostics on how recruiters read you
  • Curated company matches and warm intros
  • 48h early access to new roles

Cancel anytime.