Senior GPU Optimization Engineer

3 weeks ago


San Jose, United States Adobe Full time

Our Company

Changing the world through digital experiences is what Adobe’s all about. We give everyone—from emerging artists to global brands—everything they need to design and deliver exceptional digital experiences We’re passionate about empowering people to create beautiful and powerful images, videos, and apps, and transform how companies interact with customers across every screen.

We’re on a mission to hire the very best and are committed to creating exceptional employee experiences where everyone is respected and has access to equal opportunity. We realize that new ideas can come from everywhere in the organization, and we know the next big idea could be yours

The Opportunity
Firefly is the new family of creative generative AI models coming to Adobe products that offers a new way to conceptualize, build, and scale content. It’s a natural extension of the technology Adobe has produced over the past 40 years.

At the core of Firefly are our commercially safe AI models trained on hundreds of millions of images owned or licensed by Adobe. We are hiring for a highly strategic and visible role to apply GPU optimization skills towards improving the training efficiency and performance of these models. This is an opportunity to reach millions of creatives, helping them reinvent the way they work.

What you will be working on:

  • Interfacing with model architecture teams to co-design with hardware in mind
  • Write efficient kernels for forward and backward passes in CUDA, Cutlass / CuTe, Triton
  • Write optimized custom layers using PyTorch
  • Leverage FP8 to accelerate training and inference
  • Optimize model efficiency for Hopper/Blackwell GPU architectures
  • Write high quality, product level code that is easy to maintain and test while following standard development methodologies

What you will need to succeed:

  • B.S., M.S, or Ph.D. in Computer Science, Computer Engineering or a related area and 5+ years related experience
  • Proficiency in Linux, Docker
  • Understanding of modern transformer-based model architectures
  • Expert in Python and PyTorch
  • Understanding of distributed training fundamentals
  • Experience in C++ with an emphasis on Python integration
  • Expert in CUDA, Triton
  • Experience with Cutlass / CuTe
  • Expert in profiling – Nsight, Kineto, etc.
#J-18808-Ljbffr

  • San Jose, United States Oho Group Ltd Full time

    An industry leading smart electric vehicle company is looking for a Virtualization Engineer that specialises within GPU.Their focus areas include designing, developing, co-manufacturing, and selling high-end smart electric vehicles. They specialize within autonomous driving, digital technologies, electric powertrains, and battery systems.Roles and...


  • san jose, United States Oho Group Ltd Full time

    An industry leading smart electric vehicle company is looking for a Virtualization Engineer that specialises within GPU.Their focus areas include designing, developing, co-manufacturing, and selling high-end smart electric vehicles. They specialize within autonomous driving, digital technologies, electric powertrains, and battery systems.Roles and...

  • GPU Modeling Engineer

    6 months ago


    San Jose, United States SAMSUNG Full time

    Position Summary Samsung, a world leader in advanced semiconductor technology, is founded on a simple philosophy – the endless pursuit of excellence will create a better world for all. At Samsung Austin Research and Development Center (SARC) and Advanced Computing Lab (ACL), we are building a center of excellence for Intellectual Property (IP) that is...

  • Senior GPU Architect

    4 weeks ago


    San Diego, California, United States MediaTek Full time

    Job Title: Senior GPU ArchitectDescription: We are seeking an experienced Senior GPU Architect to join our team at MediaTek. The successful candidate will be responsible for designing and optimizing GPU clusters for industry-leading GPU hardware IP. They will collaborate with our Architecture and Software teams to develop cluster-level HW specifications that...


  • San Jose, United States Oho Group Ltd Full time

    We are working with a leading innovator in smart electric vehicles who are seeking GPU Virtualization Engineers. The company specializes in autonomous driving, digital systems, electric powertrains, and batteries. Notable advancements include battery swapping technology, Battery as a Service (BaaS), and Autonomous Driving as a Service (ADaaS). Its diverse...


  • San Diego, California, United States Qualcomm Full time

    Job Title: Senior Staff Graphics Software EngineerCompany: Qualcomm Technologies, Inc.Job Area: Engineering Group, Engineering Group > Graphics Software EngineeringJob Summary:As a leading technology innovator, Qualcomm pushes the boundaries of what's possible to enable next-generation gaming, XR, and AI experiences. Our team of skilled engineers works...


  • San Jose, United States AMD 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 Oho Group Ltd Full time

    We are working with a leading innovator in smart electric vehicles who are seeking GPU Virtualization Engineers. The company specializes in autonomous driving, digital systems, electric powertrains, and batteries. Notable advancements include battery swapping technology, Battery as a Service (BaaS), and Autonomous Driving as a Service (ADaaS). Its diverse...


  • 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...

  • GPU Design Engineer

    4 weeks ago


    San Diego, California, United States MediaTek Full time

    Job Title: GPU Design Verification EngineerCompany: MediaTekAt MediaTek, we are seeking a skilled GPU Design Verification Engineer to join our team. As a leading fabless semiconductor company, we empower innovation and inspire people to expand their horizons through smart technology.Responsibilities:Develop a deep understanding of GPU specs, including 3D...


  • San Jose, California, United States SAMSUNG Full time

    Position SummarySamsung, a world leader in advanced semiconductor technology, is founded on a simple philosophy – the pursuit of excellence for a better world. At Samsung Austin Research and Development Center (SARC) and Advanced Computing Lab (ACL), we are building a center of excellence for Intellectual Property (IP) applied to high-performance computing...


  • 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 SAMSUNG Full time

    Position Summary Samsung, a world leader in advanced semiconductor technology, is founded on a simple philosophy – the endless pursuit of excellence will create a better world for all. At Samsung Austin Research and Development Center (SARC) and Advanced Computing Lab (ACL), we are building a center of excellence for Intellectual Property (IP) that is...


  • 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 Software Guidance and Assistance, Inc. Full time

    Software Guidance & Assistance, Inc., (SGA), is searching for a GPU Software Developer - C++ for a Contract assignment with one of our premier SaaS clients in San Jose, CA or Seattle, WA . Responsibilities : Work on developing GPU components for the video processing pipeline Work on architecting, coding and productizing the high-performance GPU...


  • San Jose, United States Software Guidance and Assistance, Inc. Full time

    Software Guidance & Assistance, Inc., (SGA), is searching for a GPU Software Developer - C++ for a Contract assignment with one of our premier SaaS clients in San Jose, CA or Seattle, WA . Responsibilities : Work on developing GPU components for the video processing pipeline Work on architecting, coding and productizing the high-performance GPU components...


  • san jose, United States Oho Group Ltd Full time

    An industry leading smart electric vehicle company is looking for a Virtualization Engineer that specialises within GPU.Their focus areas include designing, developing, co-manufacturing, and selling high-end smart electric vehicles. They specialize within autonomous driving, digital technologies, electric powertrains, and battery systems.Roles and...


  • San Jose, United States Oho Group Ltd Full time

    An industry leading smart electric vehicle company is looking for a Virtualization Engineer that specialises within GPU.Their focus areas include designing, developing, co-manufacturing, and selling high-end smart electric vehicles. They specialize within autonomous driving, digital technologies, electric powertrains, and battery systems.Roles and...