AMD ยท 2 months ago
Director Machine Learning Engineering - AI/ML Model Compiler and Applications
San Jose, California
Full-time
Onsite
Director/Executive
$203K/yr - $348K/yrAdvanced Micro Devices, Inc (AMD) is committed to building innovative products that enhance computing experiences across various domains. The role involves leading a team focused on developing novel methods for efficient Generative AI and ML compiler solutions to improve training efficiency and performance on AMD platforms.
AI InfrastructureArtificial Intelligence (AI)Cloud ComputingComputerEmbedded SystemsGPUHardwareSemiconductor
Responsibilities
Lead a growing team focused on research, design, and implement novel methods for efficient Generative AI
Experience in developing ML compilers for low power inference architecture
Experience in all 3 stages of ML compiler- front end, middle end and backend
Experience in defining optimized tiling algorithms, fusion algorithms at different memory hierarchy
Influence the direction of AMD AI platform
Collaborate across teams with various groups and stakeholders
Propose and apply innovative model quantization, sparsity, and acceleration algorithms on various AMD platform, e.g., GPU/CPU/AIE
Collaborate with software and hardware team to E2E co-optimize model performance
Work with open-source framework and community (e.g., Pytorch, Huggingface) to integrate AMD optimized models and libraries
Qualification
Required
Experience with delivering ML compiler solutions
Understanding of NPU hardware architecture
Defining scalable algorithms that enables efficient problem mapping
Interfacing with customers to understand the requirements
Driving a team to develop solutions
Lead a growing team focused on research, design, and implement novel methods for efficient Generative AI
Experience in developing ML compilers for low power inference architecture
Experience in all 3 stages of ML compiler- front end, middle end and backend
Experience in defining optimized tiling algorithms, fusion algorithms at different memory hierarchy
Influence the direction of AMD AI platform
Collaborate across teams with various groups and stakeholders
Propose and apply innovative model quantization, sparsity, and acceleration algorithms on various AMD platform, e.g., GPU/CPU/AIE
Collaborate with software and hardware team to E2E co-optimize model performance
Work with open-source framework and community (e.g., Pytorch, Huggingface) to integrate AMD optimized models and libraries
A PhD or Master's degree in Computer Science, Artificial Intelligence, Machine Learning, or a related field
Preferred
Experience with ML frameworks such as Onnx, PyTorch, or TensorFlow
Experience with distributed training and distributed training frameworks, such as DeepSpeed
Experience with LLMs or computer vision, especially large models, is a plus
Excellent C++ and Python programming skills, including debugging, profiling, and performance analysis
Experience with ML pipelines
Strong communication and problem-solving skills
Strong leadership and management skills
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
GlobeNewswire
2026-01-09
Company data provided by crunchbase