Software Engineer, Triton Compiler

3 weeks ago


San Francisco, United States OpenAI Full time

About the Team

Our mission at OpenAI is to discover and enact the path to safe, beneficial AGI. To do this, we believe that many technical breakthroughs are needed in generative modeling, reinforcement learning, large scale optimization, active learning, among other topics.

About the Role

As a Software Engineer, you will help build AI systems that can perform previously impossible tasks or achieve outstanding levels of performance. This requires good engineering (for example designing, implementing, and optimizing state-of-the-art AI models), writing bug-free machine learning code (surprisingly difficult), and building the science behind the algorithms employed. In all the projects this role pursues, the ultimate goal is to push the field forward.

The Research Acceleration team builds high-quality research tools and frameworks to increase research productivity across OpenAI, with the goal of accelerating progress towards AGI. For example, we develop Triton, a language and compiler for writing custom GPU kernels. The aim of Triton is to provide an open-source environment to write fast code at higher productivity than CUDA.

We frequently collaborate with other teams to speed up the development of new state-of-the-art capabilities. For example, we recently collaborated with our Codegen research team on the Codex model, which can generate code in Python and many other languages.

Do you love research tools, compilers, and collaborating on cutting-edge AI models? If so, this role is for you We are looking for people who are self-directed and enjoy determining the most meaningful problem to solve in order to accelerate our research.

We're looking for a track record of:

  1. 3+ years of relevant engineering experience
  2. Owning problems end-to-end, with a willingness to pick up whatever knowledge is missing to get the job done
  3. Bonus: contributions to an AI framework such as PyTorch or Tensorflow, or compilers such as GCC, LLVM, or MLIR

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 do not discriminate on the basis of race, religion, national origin, gender, sexual orientation, age, veteran status, disability or any other legally protected status.

For US Based Candidates: Pursuant to the San Francisco Fair Chance Ordinance, we will consider qualified applicants with arrest and conviction records.

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

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.

#J-18808-Ljbffr

  • San Francisco, United States Gensyn Full time

    Machine intelligence will soon take over humanity’s role in knowledge-keeping and creation. What started in the mid-1990s as the gradual off-loading of knowledge and decision making to search engines will be rapidly replaced by vast neural networks - with all knowledge compressed into their artificial neurons. Unlike organic life, machine intelligence,...


  • san francisco, United States Walter Bacon, LLC Full time

    Sr. Compiler Software Engineer (AI DevTool Startup)ABOUT US:AI DevTool for Refactoring Enterprise Codebases at Scale in SecondsFounder and CEO with a Previous AI Startup Successful ExitHQ in San Francisco. Hybrid in-office / remote$15M Seed FundingIDEAL CANDIDATE PROFILE:10+ years of Software Engineering4+ years of Compiler EngineeringDesire to work at a...


  • San Francisco, United States Walter Bacon, LLC Full time

    Sr. Compiler Software Engineer (AI DevTool Startup)ABOUT US:AI DevTool for Refactoring Enterprise Codebases at Scale in SecondsFounder and CEO with a Previous AI Startup Successful ExitHQ in San Francisco. Hybrid in-office / remote$15M Seed FundingIDEAL CANDIDATE PROFILE:10+ years of Software Engineering4+ years of Compiler EngineeringDesire to work at a...


  • san francisco, United States Walter Bacon, LLC Full time

    Sr. Compiler Software Engineer (AI DevTool Startup)ABOUT US:AI DevTool for Refactoring Enterprise Codebases at Scale in SecondsFounder and CEO with a Previous AI Startup Successful ExitHQ in San Francisco. Hybrid in-office / remote$15M Seed FundingIDEAL CANDIDATE PROFILE:10+ years of Software Engineering4+ years of Compiler EngineeringDesire to work at a...


  • San Jose, California, United States Advanced Micro Devices Full time

    We are at the forefront of innovation, driving technological advancements that enrich our industry, communities, and world. Our mission is to build great products that accelerate next-generation computing experiences.Our core team is developing a cutting-edge machine learning model compiler targeting AMD Inference Accelerator AIE hardware devices. The...


  • San Jose, United States Advanced Micro Devices , Inc. Full time

    WHAT YOU DO AT AMD CHANGES EVERYTHING We care deeply about transforming lives with AMD technology to enrich our industry, our communities, and the world. Our mission is to build great products that accelerate next-generation computing experiences the building blocks for the data center, artificial intelligence, PCs, gaming and embedded. Underpinning our...


  • San Jose, United States Advanced Micro Devices , Inc. Full time

    WHAT YOU DO AT AMD CHANGES EVERYTHING We care deeply about transforming lives with AMD technology to enrich our industry, our communities, and the world. Our mission is to build great products that accelerate next-generation computing experiences – the building blocks for the data center, artificial intelligence, PCs, gaming and embedded. Underpinning our...


  • San Jose, United States Advanced Micro Devices , Inc. Full time

    WHAT YOU DO AT AMD CHANGES EVERYTHING We care deeply about transforming lives with AMD technology to enrich our industry, our communities, and the world. Our mission is to build great products that accelerate next-generation computing experiences – the building blocks for the data center, artificial intelligence, PCs, gaming and embedded. Underpinning our...


  • San Francisco, United States Bodo Inc Full time

    At Bodo, we are redefining what's possible with Python for analytics, data engineering, and AI. We've developed the world's first auto-parallelizing Python compiler, delivering HPC-level performance and efficiency for large-scale data processing. Now, we are building upon this foundational software and applying it directly to analytics and AI, empowering...


  • San Francisco, United States Bodo Inc Full time

    At Bodo, we are redefining what's possible with Python for analytics, data engineering, and AI. We've developed the world's first auto-parallelizing Python compiler, delivering HPC-level performance and efficiency for large-scale data processing. Now, we are building upon this foundational software and applying it directly to analytics and AI, empowering...


  • San Francisco, California, United States OpenAI Full time

    About the RoleWe are seeking a skilled AI Research Tools Engineer to join our Research Acceleration team. As a key member of our team, you will play a crucial role in building high-quality research tools and frameworks to accelerate progress towards Artificial General Intelligence (AGI).Your primary focus will be on developing Triton, a language and compiler...


  • San Jose, United States Advanced Micro Devices , Inc. Full time

    WHAT YOU DO AT AMD CHANGES EVERYTHING We care deeply about transforming lives with AMD technology to enrich our industry, our communities, and the world. Our mission is to build great products that accelerate next-generation computing experiences the building blocks for the data center, artificial intelligence, PCs, gaming and embedded. Underpinning our...


  • San Francisco, CA, United States Walter Bacon, LLC Full time

    Sr. Compiler Software Engineer (AI DevTool Startup)ABOUT US:AI DevTool for Refactoring Enterprise Codebases at Scale in SecondsFounder and CEO with a Previous AI Startup Successful ExitHQ in San Francisco. Hybrid in-office / remote$15M Seed FundingIDEAL CANDIDATE PROFILE:10+ years of Software Engineering4+ years of Compiler EngineeringDesire to work at a...


  • San Francisco, United States Nexus Full time

    We are seeking a highly skilled Senior Compiler Engineer to join our team and play a critical role in pushing the boundaries of performance and efficiency for our decentralized platform. You will leverage your expertise in LLVM and Cranelift to design and implement cutting-edge compiler optimizations for various architectures as we innovate in the world of...


  • San Jose, United States Advanced Micro Devices Full time

    WHAT YOU DO AT AMD CHANGES EVERYTHINGWe care deeply about transforming lives with AMD technology to enrich our industry, our communities, and the world. Our mission is to build great products that accelerate next-generation computing experiences – the building blocks for the data center, artificial intelligence, PCs, gaming and embedded. Underpinning our...


  • San Jose, United States Advanced Micro Devices, Inc. Full time

    WHAT YOU DO AT AMD CHANGES EVERYTHINGWe care deeply about transforming lives with AMD technology to enrich our industry, our communities, and the world. Our mission is to build great products that accelerate next-generation computing experiences - the building blocks for the data center, artificial intelligence, PCs, gaming and embedded. Underpinning our...


  • San Jose, United States AMD Full time

    WHAT YOU DO AT AMD CHANGES EVERYTHING We care deeply about transforming lives with AMD technology to enrich our industry, our communities, and the world. Our mission is to build great products that accelerate next-generation computing experiences – the building blocks for the data center, artificial intelligence, PCs, gaming and embedded. Underpinning our...


  • San Jose, California, United States Advanced Micro Devices , Inc. Full time

    About UsAt Advanced Micro Devices, Inc., we're revolutionizing the world of technology with our cutting-edge products. Our mission is to accelerate next-generation computing experiences, and we're looking for talented individuals to join our team.Job DescriptionWe're seeking an experienced Machine Learning Compiler Engineer to develop a high-performance...


  • San Francisco, CA, United States Bodo, Inc. Full time

    At Bodo, we are redefining what’s possible with Python for analytics, data engineering, and AI. We’ve developed the world’s first auto-parallelizing Python compiler, delivering HPC-level performance and efficiency for large-scale data processing. Now, we are building upon this foundational software and applying it directly to analytics and AI,...


  • San Jose, California, United States AMD Full time

    **About AMD**We are a leading technology company dedicated to transforming lives with our innovative products. Our mission is to accelerate next-generation computing experiences and push the limits of innovation to solve the world's most important challenges.**Job Summary**We are seeking an experienced engineer to join our core team working on developing a...