Deep Learning Compiler Software Engineer for Advanced Micro Devices
2 weeks ago
We are seeking a highly skilled Machine Learning (ML) Compiler Software Engineer to join our team at Advanced Micro Devices, Inc. (AMD). As a key member of our AI group, you will play a crucial role in developing software tools to deploy cutting-edge ML models on AMD's Neural Processing Units (NPU).
Key Responsibilities- Design and implement compiler optimization passes that translate Gen-AI ML inference models like SDXL-Turbo, Llama2, Mistral, etc. into low-level code for compute and dataflow on NPUs.
- Collaborate with Architecture and runtime software teams to develop optimization strategies for the compiler.
- Implement loop tiling, and buffer allocation strategies for efficient compute and DRAM access.
- Develop back-end optimization passes to convert high-level representation into driver calls for different NPU generations.
- Testing and validation of optimized kernels and dataflow graphs using performance models.
- Develop and maintain unit tests and integration tests for the compiler to support different generations of HW architectures.
- Enable detailed profiling and debugging tools for analyzing performance bottlenecks and deadlocks in the dataflow schemes.
- Strong understanding of ML compiler optimizations (front-end, middle-end, back-end).
- Experience with machine learning frameworks (e.g., TensorFlow, PyTorch).
- Experience working with on GPUs, TPUs, NPUs, or vector processors is a must.
- Experience with ML models such as CNN, LSTM, LLMs.
- Excellent programming skills in Python, C++, or similar languages.
- A passion for innovation and a strong desire to push the boundaries of machine learning performance.
- Master's degree or PhD. in Computer Science, Engineering, or a related field (or Bachelor's degree with significant experience).
At AMD, we are committed to building great products that accelerate next-generation computing experiences. Our mission is to transform lives with AMD technology to enrich our industry, our communities, and the world. We strive for execution excellence while being direct, humble, collaborative, and inclusive of diverse perspectives.
We are an equal opportunity, inclusive employer and will consider all applicants without regard to age, ancestry, color, marital status, medical condition, mental or physical disability, national origin, race, religion, political and/or third-party affiliation, sex, pregnancy, sexual orientation, gender identity, military or veteran status, or any other characteristic protected by law.
-
Senior Compiler Engineer for Deep Learning
10 hours ago
San Jose, California, United States Advanced Micro Devices, Inc Full timeAbout the RoleWe are seeking a highly skilled Senior Compiler Engineer to join our team at Advanced Micro Devices, Inc. as a Deep Learning Compiler Engineer. In this role, you will be responsible for enhancing the LLVM compilation on current and future AMD GPUs.Key ResponsibilitiesSupport and contribute to AMD GPU backend compilation in LLVM.Understand...
-
San Jose, California, United States Advanced Micro Devices, Inc Full timeAbout the RoleWe are seeking a highly skilled Machine Learning Compiler Software Engineer to join our team at Advanced Micro Devices, Inc. and play a crucial role in developing software toolsets to deploy cutting-edge Machine Learning models on our XDNA Neural Processing Units (NPU).Key ResponsibilitiesDesign, implement, and optimize compilers that translate...
-
San Jose, California, United States Advanced Micro Devices, Inc Full timeAbout the RoleWe are seeking a highly skilled Machine Learning (ML) Compiler Software Engineer to join our team in the AI group at Advanced Micro Devices, Inc. (AMD). As a key member of our team, you will play a crucial role in developing software toolsets to deploy cutting-edge ML models on AMD's XDNA Neural Processing Units (NPU).Key...
-
Deep Learning Compiler Engineer for Ryzen AI NPU
2 months ago
San Jose, United States Advanced Micro Devices , Inc. Full timeWHAT 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...
-
Deep Learning Compiler Engineer for Ryzen AI NPU
3 weeks ago
San Jose, United States Advanced Micro Devices , Inc. Full timeWHAT 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, California, United States Advanced Micro Devices , Inc. Full timeJOIN AMD IN SHAPING THE FUTURE OF TECHNOLOGY At AMD, we are dedicated to enhancing lives through our innovative technology, impacting our industry, communities, and the world at large. Our goal is to create exceptional products that propel next-generation computing experiences, serving as the foundation for data centers, artificial intelligence, personal...
-
San Jose, California, United States Advanced Micro Devices, Inc Full timeAbout the RoleWe are seeking a highly skilled Senior Machine Learning Performance Optimization Engineer to join our team at Advanced Micro Devices, Inc. As a key member of our engineering team, you will be responsible for optimizing the performance of machine learning workloads on our hardware platforms.Key ResponsibilitiesDevelop and implement performance...
-
Principal Machine Learning Compiler Engineer
2 weeks ago
San Jose, United States Advanced Micro Devices, Inc. Full timeWHAT 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...
-
GPU Design Verification Engineer
4 days ago
San Diego, California, United States Advanced Micro Devices, Inc Full timeAbout the RoleWe are seeking a highly skilled GPU Design Verification Engineer to join our team at Advanced Micro Devices, Inc. (AMD). As a GPU Design Verification Engineer, you will play a critical role in ensuring the quality and functionality of our graphics processing units (GPUs).Key ResponsibilitiesWork closely with the hardware and firmware team to...
-
ML Compiler Software Development Engineer
1 month ago
San Jose, California, United States Advanced Micro Devices Full timeWHAT 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...
-
Data Analyst Intern/Co-Op
1 day ago
Santa Clara, California, United States Advanced Micro Devices, Inc Full timeAbout the RoleWe are seeking a highly motivated Data Analyst intern/co-op to join our team at Advanced Micro Devices, Inc. and contribute to our next-generation computing experience.Key ResponsibilitiesAnalyze and organize raw data to craft proofs of concepts, prototype, and implement solutions using relevant analysis methods and visualization...
-
Senior Infinity Fabric Verification Engineer
20 hours ago
Santa Clara, California, United States Advanced Micro Devices, Inc Full timeAbout the RoleWe are seeking a highly skilled Senior Verification Engineer to join our Infinity Fabric transport layer verification team at Advanced Micro Devices, Inc. As a Senior Verification Engineer, you will play a critical role in verifying our configurable switches and die-to-die interconnect.Key ResponsibilitiesCollaborate with architects, hardware...
-
Senior Infinity Fabric Verification Engineer
5 days ago
Santa Clara, California, United States Advanced Micro Devices, Inc Full timeAbout the RoleWe are seeking a highly skilled Senior Infinity Fabric Verification Engineer to join our team at Advanced Micro Devices, Inc. As a key member of our verification team, you will be responsible for verifying our configurable switches and die-to-die interconnect.Key ResponsibilitiesCollaborate with architects, hardware engineers, and firmware...
-
San Jose, United States Advanced Micro Devices, Inc. Full timeWHAT 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...
-
Machine Learning Engineer Intern
10 hours ago
Santa Clara, California, United States Advanced Micro Devices, Inc Full timeJob Summary:Advanced Micro Devices, Inc. is seeking a highly motivated Machine Learning Engineer Intern to join our team and participate in research and development of next-generation product differentiation features with extraordinary ML/AI engineers.Key Responsibilities:Assist in the development of machine learning algorithms and models to improve AI...
-
Senior AI Infrastructure Software Engineer
20 hours ago
San Jose, California, United States Advanced Micro Devices, Inc Full timeAbout the RoleWe are seeking a highly skilled Senior Software Engineer to join our team at Advanced Micro Devices, Inc. as a key member of our AI Infrastructure team. The successful candidate will be responsible for contributing to the development and release of our inference engine, which will enable our customers to leverage high-performance AI models on...
-
Machine Learning Engineer Intern
6 days ago
Santa Clara, California, United States Advanced Micro Devices, Inc Full timeJob DescriptionJob Title: Machine Learning Engineer InternJob Summary:We are seeking a highly motivated and talented Machine Learning Engineer Intern to join our team and contribute to the development of next-generation product differentiation features. As a Machine Learning Engineer Intern, you will have the opportunity to work with our team of experienced...
-
Senior AI Infrastructure Software Engineer
3 days ago
San Jose, California, United States Advanced Micro Devices Full timeAbout the RoleWe are seeking a highly skilled Senior AI Infrastructure Software Engineer to join our team at Advanced Micro Devices (AMD). As a key member of our infrastructure team, you will play a critical role in the development and release of our inference engine, which will enable our customers to leverage high-performance AI models on top of AMD's...
-
San Jose, California, United States Advanced Micro Devices, Inc. Full timeAbout the RoleWe are seeking a highly skilled Principal Machine Learning Performance Optimization Specialist to join our team at Advanced Micro Devices, Inc. (AMD). As a key member of our organization, you will play a critical role in shaping the future of AI acceleration and performance optimization.Key ResponsibilitiesPerformance Modeling and Analysis:...
-
Lead Systems Design Engineer
14 hours ago
San Jose, California, United States Advanced Micro Devices, Inc Full timeAbout the RoleWe are seeking a highly skilled and experienced Lead Systems Design Engineer to join our team at Advanced Micro Devices, Inc. This is a key role that will play a critical part in the development of test systems and hardware for the characterization of our advanced device families.Key ResponsibilitiesEnhance existing test software for...
