Advanced Micro Devices, Inc

Sr.. Software Development Engineer - GPU Kernel Development

Advanced Micro Devices, Inc • $140K — $180K *
Santa Clara, CA 95051In-Person
Enterprise Technology
Less than 5 years of experience
Today
Be an Early Applicant
Job Overview by Ladders

Qualifications

  • 5-7 years of experience in GPU development and optimization.
  • Expertise in C++ and Python programming.
  • Strong background in Linux environments for high-performance computing.
  • Experience with deep learning frameworks like TensorFlow and PyTorch.
  • Proficiency in GPU kernels and low-level programming for AI applications.

Responsibilities

  • Optimize deep learning frameworks for AMD GPUs.
  • Develop and enhance performance of GPU kernels for AI operations.
  • Design and optimize deep learning models for AMD GPU performance.
  • Collaborate with internal GPU teams to improve training and inference with AMD GPUs.
  • Engage with open-source maintainers for code integration and compliance.
  • Work on optimizing deep learning performance in distributed computing setups.
  • Apply cutting-edge compiler technologies to enhance performance.
  • Mentor and guide junior team members in best engineering practices.

Benefits

  • Comprehensive benefits package including health insurance.
  • Opportunities for professional development and career growth.
  • Flexible working environment and schedule.
  • Access to cutting-edge technologies and tools.
  • Work on impactful, large-scale AI projects.
Full Job Description
THE ROLE:

As a core member of the team, you will play a pivotal role in optimizing and developing deep learning frameworks for AMD GPUs. Your expertise will be critical in enhancing GPU kernels, deep learning models, and training/inference performance across multi-GPU and multi-node systems. You will engage with both internal GPU library teams and open-source maintainers to ensure seamless integration of optimizations, utilizing cutting-edge compiler technologies and advanced engineering principles to drive continuous improvement.

THE PERSON:

Seeking an Industry Leading Expert C++ developer with advanced technical and analytical skills in Linux environments. The ideal candidate will excel in providing technical leadership, guiding teams, and driving projects/initiatives independently. You will define goals, scope, and own development efforts while collaborating effectively within a high-performing team.

KEY RESPONSIBILITIES:
  • Optimize Deep Learning Frameworks: Enhance and optimize frameworks like TensorFlow and PyTorch for AMD GPUs in open-source repositories.
  • Develop GPU Kernels: Create and optimize GPU kernels to maximize performance for specific AI operations.
  • Develop & Optimize Models: Design and optimize deep learning models specifically for AMD GPU performance.
  • Collaborate with GPU Library Teams: Work closely with internal teams to analyze and improve training and inference performance on AMD GPUs.
  • Collaborate with Open-Source Maintainers: Engage with framework maintainers to ensure code changes are aligned with requirements and integrated upstream.
  • Work in Distributed Computing Environments: Optimize deep learning performance on both scale-up (multi-GPU) and scale-out (multi-node) systems.
  • Utilize Cutting-Edge Compiler Tech: Leverage advanced compiler technologies to improve deep learning performance.
  • Optimize Deep Learning Pipeline: Enhance the full pipeline, including integrating graph compilers.
  • Software Engineering Best Practices: Apply sound engineering principles to ensure robust, maintainable solutions.
  • Lead, Guide & Mentor: Provide strategic direction and mentorship to junior team members, fostering growth and collaboration through code reviews, knowledge sharing, and technical guidance.


PREFERRED EXPERIENCE:
  • GPU Kernel Development & Optimization: Deep expertise in designing and optimizing GPU kernels for deep learning on AMD GPUs using HIP, CUDA, and assembly (ASM). Strong knowledge of AMD architectures (GCN, RDNA) and low-level programming to maximize performance for AI operations, leveraging tools like Compute Kernel (CK), CUTLASS, and Triton for multi-GPU and multi-platform performance.
  • Deep Learning Integration: Proven ability and experience to integrate GPU-accelerated compute into ML frameworks (e.g., PyTorch, TensorFlow), with a focus on throughput, scalability, and efficient execution for training and inference workloads.
  • Software Engineering Excellence: Advanced proficiency in Python and C++ with deep experience in performance tuning, debugging, and robust test design, ensuring reliable, maintainable, high-performance codebases.
  • High-Performance Computing: Broad and indepth experience with large-scale, heterogeneous compute environments; adept at optimizing AI workloads for performance, efficiency, and resource utilization across clusters.
  • Compiler Optimization: Thorough and detailed understanding of compiler internals, LLVM, and ROCm, with the ability to drive system-level optimizations from source to machine code.


ACADEMIC CREDENTIALS:
  • Master's and/ PhD degree in Computer Science, Computer Engineering, Electrical Engineering, or a related field.


#LI-JG1

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 Enterprise Technology Jobs

Find similar Sr.. Software Development Engineer - GPU Kernel Development jobs:

NationwideSanta Clara, CA