Advanced Micro Devices, Inc

Triton Compiler and Kernel Software Engineer

Advanced Micro Devices, Inc • $150K — $180K *
San Jose, CA 95123In-Person
Consumer Technology
Less than 5 years of experience
Today
Be an Early Applicant
Job Overview by Ladders

Qualifications

  • 5-7 years of experience in GPU architecture and programming.
  • Strong background in compiler development and optimization.
  • Experience with high-performance GPU kernels.
  • Familiarity with multi-GPU communication and collective operations.
  • Knowledge in distributed AI training and inference techniques.
  • Proficient in low-level performance analysis across different components.
  • Bachelor's or Master's degree in relevant fields (Computer Science, Computer Engineering, Electrical Engineering).

Responsibilities

  • Develop and optimize Triton compiler support for AMD GPUs.
  • Improve compiler lowering, optimization, scheduling, and code generation processes.
  • Create high-performance kernels for various AI workloads such as attention and GEMM.
  • Optimize multi-GPU kernels and develop communication primitives.
  • Enable new AMD GPU capabilities through Triton abstractions.
  • Analyze performance metrics, including compute and memory usage.
  • Resolve complex correctness and performance issues in both single- and multi-GPU workloads.
  • Collaborate with teams across GPU architecture and AI frameworks.
  • Contribute to upstream Triton and LLVM/MLIR designs and implementations.
  • Provide mentorship and technical leadership within the team.

Benefits

  • Comprehensive health and wellness plans.
  • Flexible work arrangements with hybrid options.
  • Opportunities for professional development and continuous learning.
  • Participation in AMD's growth-focused initiatives.
  • Engagement in open-source contributions and community projects.
Full Job Description
THE ROLE:

We are seeking a Senior Triton Compiler and Kernel Engineer to advance Triton performance and capabilities on AMD GPUs.

Triton is an open-source language and compiler for developing high-performance GPU kernels in Python. It is a critical layer in the AI software stack, connecting frameworks and workloads to GPU hardware. Triton is strategic to AMD's AI roadmap, and AMD is investing fully in making it a first-class platform for current and future AMD GPUs.

You will work across GPU architecture, compilers, kernels, multi-GPU communication, and AI frameworks while contributing to upstream Triton and AMD's ROCm software stack.

THE PERSON:

The ideal candidate has strong experience in several of the following areas:
  • GPU architecture and programming
  • Compiler development and optimization
  • High-performance GPU kernels
  • Multi-GPU communication and collective operations
  • Distributed AI training and inference
  • AI workload and framework performance
  • Low-level performance analysis

You can reason across the stack-from distributed AI algorithms and Triton programs to compiler transformations, communication libraries, generated instructions, and GPU hardware.

KEY RESPONSIBILITIES:
  • Develop and optimize Triton compiler support for AMD GPUs.
  • Improve compiler lowering, optimization, scheduling, and code generation.
  • Create high-performance kernels for attention, GEMM, MoE, and other AI workloads.
  • Develop and optimize multi-GPU kernels and communication primitives.
  • Enable new AMD GPU and interconnect capabilities through effective Triton abstractions.
  • Analyze compute, memory, communication, occupancy, register usage, and generated code.
  • Resolve complex correctness and performance issues across single- and multi-GPU workloads.
  • Collaborate with GPU architecture, ROCm, PyTorch, and AI framework teams.
  • Contribute designs and implementations to upstream Triton and LLVM/MLIR.
  • Provide technical leadership, code reviews, and mentoring.


PREFERRED EXPERIENCE:
  • Experience with AMD GPU architecture and ROCm is highly desirable.
  • Experience optimizing kernels with Triton, HIP, CUDA, or GPU assembly.
  • Experience with Triton, LLVM, MLIR, or another optimizing compiler.
  • Knowledge of GPU execution models, memory hierarchies, synchronization, and instruction pipelines.
  • Experience with collective communication, distributed programming, and libraries such as RCCL or NCCL.
  • Understanding of communication topologies, interconnects, synchronization, and communication-computation overlap.
  • Familiarity with distributed training, inference, tensor parallelism, expert parallelism, or pipeline parallelism.
  • Experience using AI coding tools and autonomous agents to accelerate software development, debugging, benchmarking, and kernel performance tuning.
  • Familiarity with AI primitives, reduced-precision formats, and performance profiling.
  • Contributions to complex or open-source software projects.
  • Strong analytical, debugging, communication, and collaboration skills.


ACADEMIC CREDENTIALS:
  • Bachelor's or Master's degree in Computer Science, Computer Engineering, Electrical Engineering, or equivalent


This role is not eligible for visa sponsorship.

#LI-G11

#LI-HYBRID

Benefits offered are described: AMD benefits at a glance.

About Advanced Micro Devices, Inc

Advanced Micro Devices, Inc. Careers

Join the innovative forefront of technology with a career at Advanced Micro Devices, Inc. (AMD), a leader in semiconductor development. As part of our global team, you will contribute to an organization renowned for its dedication to innovation, leadership, and diversity in the tech industry.

Work You’ll Do

At AMD, we offer job opportunities that push the boundaries of what is possible. Our team is composed of professionals who lead the way in microprocessor and graphics technology, driving industry standards and innovation. With AMD, you will be part of a culture that values growth and professional development, ensuring that every team member has the opportunity to excel.

Transform Your Career

AMD is not just about advancing technology, but also about advancing careers. Whether you are looking for an internship, a full-time position, or leadership roles, AMD provides the platform to propel your career to new heights. Our commitment to professional growth is matched by our dedication to diversity and inclusion, making AMD a place where everyone can thrive.

Innovative Work Environment

Join a team of over 12,000 dedicated professionals at the intersection of technology, industry expertise, and digital innovation. At AMD, you will work on groundbreaking projects that shape the future of computing and graphics. Our collaborative environment encourages networking and the sharing of ideas across teams and disciplines.

Career Development and Benefits

AMD is committed to the development of its employees. We offer robust training programs, including leadership development and diversity training, to ensure our team is equipped for both current challenges and future opportunities. Our benefits package is designed to support the well-being and financial security of our employees and their families.

Explore Job Opportunities

From engineering to marketing, AMD offers a range of career paths that cater to diverse skills and interests. Our hiring process is designed to be transparent and engaging, helping you to understand where you fit within our team and how you can contribute to our collective goals.

Stay Connected

Join Our Team Search open positions that match your skills and interest. We look for passionate, curious, creative, and solution-driven team players. Explore the opportunities to join a company that’s committed to your career growth and to innovation in the technology sector.

Keep Up to Date

Stay ahead with career tips, insider perspectives, and industry-leading insights you can put to use today—all from the people who work here.

Job Alert Emails

Personalize your subscription to receive job alerts, latest news, and insider tips tailored to your preferences. Discover the exciting and rewarding career opportunities that await at Advanced Micro Devices, Inc.

Interview and Resume Tips

Prepare for your future with AMD by accessing resources that help you craft your resume and excel in interviews. Our goal is to help you showcase your best professional self and align your skills with the needs of our dynamic team. At Advanced Micro Devices, Inc., we empower our employees to innovate, lead, and grow. Join us in driving the future of technology while building a rewarding and sustainable career.
Learn more about Advanced Micro Devices, Inc
Size
15,500 employees
Market Cap
$100.9 billion
Industry
Manufacturing & Automotive
Net Income
$2.4 billion
Founded
1969
5 Year Trend
+30.9%
Revenue
$9.7 billion
NASDAQ
AMD
* Ladders Estimates

Similar Jobs

Ad banner background

Get Ready For Your
Next Interview

More Jobs at Advanced Micro Devices, Inc

More Consumer Technology Jobs

Find similar Triton Compiler and Kernel Software Engineer jobs:

NationwideSan Jose, CA