AMD ยท 1 week ago
Triton AI Compiler Engineer
Santa Clara, CA
Full-time
Onsite
Mid, Senior Level
$163K/yr - $245K/yrAMD is a company dedicated to building innovative products that enhance next-generation computing experiences. The role involves developing and optimizing Deep Learning applications and AI frameworks for AMD GPU compute platforms, collaborating with teams to improve performance and efficiency.
AI InfrastructureArtificial Intelligence (AI)Cloud ComputingComputerEmbedded SystemsGPUHardwareSemiconductor
Responsibilities
Optimize Deep Learning Frameworks: In depth experience in 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 tightly 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
Qualification
Required
Strong technical and analytical skills in C++ development in a Linux environment
Ability to work as part of a team, while also being able to work independently, define goals and scope and lead your own development effort
In depth experience in enhance and optimize frameworks like TensorFlow and PyTorch for AMD GPUs in open-source repositories
Create and optimize GPU kernels to maximize performance for specific AI operations
Design and optimize deep learning models specifically for AMD GPU performance
Work tightly with internal teams to analyze and improve training and inference performance on AMD GPUs
Engage with framework maintainers to ensure code changes are aligned with requirements and integrated upstream
Optimize deep learning performance on both scale-up (multi-GPU) and scale-out (multi-node) systems
Leverage advanced compiler technologies to improve deep learning performance
Enhance the full pipeline, including integrating graph compilers
Apply sound engineering principles to ensure robust, maintainable solutions
Bachelor's and/or Master's degree in Computer Science, Computer Engineering, Electrical Engineering, or a related field
Preferred
Proficient experienced 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
Proficient experienced in integrating optimized GPU performance into machine learning frameworks (e.g., TensorFlow, PyTorch) to accelerate model training and inference, with a focus on scaling and throughput
Proficient skills in Python and C++, with experience in debugging, performance tuning, and test design to ensure high-quality, maintainable software solutions
Proficient experience in running large-scale workloads on heterogeneous compute clusters, optimizing for efficiency and scalability
Solid understanding of compiler theory and tools like LLVM and ROCm for kernel and system performance optimization
Benefits
AMD benefits at a glance.
Company
AMD
Advanced Micro Devices is a semiconductor company that designs and develops graphics units, processors, and media solutions.
Founded in 1969
10001+ employees
H1B Sponsorship
AMD has a track record of offering H1B sponsorships. Please note that this does not
guarantee sponsorship for this specific role. Below presents additional info for your
reference. (Data Powered by US Department of Labor)
Distribution of Different Job Fields Receiving Sponsorship
Represents job field similar to this job
Trends of Total Sponsorships
2025 (836)
2024 (770)
2023 (551)
2022 (739)
2021 (519)
2020 (547)
Funding
Current Stage
Public CompanyTotal Funding
unknownKey Investors
OpenAIDaniel Loeb
2025-10-06Post Ipo Equity
2023-03-02Post Ipo Equity
2021-06-29Post Ipo Equity
Leadership Team
Lisa Su
Chair & CEO
Mark Papermaster
CTO and EVP
Recent News
2026-01-14
2026-01-13
Morningstar.com
2026-01-11
Company data provided by crunchbase