Current jobs related to Machine Learning Compiler Software Engineer for Advanced Micro Devices, Inc. - San Jose California - Advanced Micro Devices, Inc


  • San Jose, California, United States Advanced Micro Devices , Inc. Full time

    {"Responsibilities": "As a Machine Learning Compiler Engineer at Advanced Micro Devices, Inc., you will play a crucial role in developing software toolsets to deploy cutting-edge machine learning models on AMD's XDNA Neural Processing Units (NPUs). Your primary responsibilities will include designing, implementing, and optimizing compilers to translate...


  • San Jose, California, United States Advanced Micro Devices , Inc. Full time

    Transforming Lives with AMD TechnologyWe're on a mission to build innovative products that accelerate next-generation computing experiences. As a skilled engineer, you'll join a core team of industry specialists working on developing a cutting-edge machine learning model compiler targeting AMD Inference Accelerator AIE hardware devices.The Role:Develop a...


  • San Jose, United States Advanced Micro Devices, Inc. Full time

    WHAT 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...


  • San Jose, United States Advanced Micro Devices , Inc. Full time

    WHAT 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 time

    About the RoleWe are seeking a talented Machine Learning (ML) Compiler Software Engineer to join our growing team in the AI group and play a crucial role in developing software toolsets to deploy cutting-edge ML models on AMD's XDNA Neural Processing Units (NPU).Key ResponsibilitiesDesign and develop novel algorithms for tiling and mapping quantized ML...


  • San Jose, California, United States Advanced Micro Devices , Inc. Full time

    Transforming Lives with AMD TechnologyWe are committed to enriching our industry, communities, and the world through AMD technology. 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.The Role:We are seeking a talented...


  • San Jose, California, United States Advanced Micro Devices , Inc. Full time

    Transforming Lives with AMD TechnologyWe're passionate about using AMD technology to enrich our industry, communities, and the world. Our mission is to build innovative products that accelerate next-generation computing experiences - the building blocks for the data center, artificial intelligence, PCs, gaming, and embedded systems.At AMD, we push the limits...


  • San Jose, California, United States Advanced Micro Devices , Inc. Full time

    About Advanced Micro Devices, Inc.We are a leader in the technology industry, dedicated to transforming lives with our innovative solutions. Our mission is to build great products that accelerate next-generation computing experiences, including the data center, artificial intelligence, PCs, gaming, and embedded solutions.Our culture is built on the...


  • San Jose, California, United States Advanced Micro Devices, Inc Full time

    We are seeking a highly skilled Senior ASIC Design Engineer to join our team at Advanced Micro Devices, Inc. The ideal candidate will have a strong background in SerDes solution delivery and a passion for innovation.The successful candidate will be responsible for driving end-to-end delivery of SerDes solutions, coordinating implementation and optimization...


  • San Jose, California, United States AMD Full time

    Transforming Lives with AMD TechnologyWe're on a mission to build innovative products that accelerate next-generation computing experiences. As a skilled software development engineer, you'll join a talented team working on cutting-edge machine learning model compilers targeting AMD AIE in Ryzen AI PC and embedded FPGA devices.The RoleDevelop a compiler that...


  • San Jose, California, United States AMD Full time

    About the Role:AMD is seeking a skilled Machine Learning Compiler Engineer to join a core team of talented industry specialists working on developing a cutting-edge machine learning model compiler targeting AMD Inference Accelerator AIE hardware devices.The ideal candidate will be responsible for implementing and improving effective algorithms and techniques...


  • San Jose, California, United States AMD Full time

    Unlock the Power of Machine Learning with AMDAt AMD, we're pushing the boundaries of innovation to solve the world's most complex challenges. We're seeking a skilled Machine Learning Compiler Engineer to join our team of talented industry specialists working on developing a cutting-edge machine learning model compiler targeting AMD Inference Accelerator AIE...


  • San Jose, California, United States Advanced Micro Devices , Inc. Full time

    Transforming Lives with AMD TechnologyWe are seeking a highly skilled Senior ASIC Design Engineer to join our team at Advanced Micro Devices, Inc. Our mission is to build great products that accelerate next-generation computing experiences.The Role:As a Senior ASIC Design Engineer, you will drive end-to-end delivery of SerDes solutions, coordinating...


  • San Jose, United States Advanced Micro Devices, Inc. Full time

    Principal Machine Learning Performance EngineerWe 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....


  • San Jose, California, United States Advanced Micro Devices , Inc. Full time

    Transforming Lives with AMD TechnologyWe're on a mission to build innovative products that accelerate next-generation computing experiences. As a Principal Machine Learning Performance Engineer, you'll play a crucial role in shaping the future of AI acceleration.Key Responsibilities:Develop and improve our framework, tools, and infrastructure for performance...


  • San Jose, California, United States Advanced Micro Devices , Inc. Full time

    We are seeking a highly skilled Silicon Design Engineer to join our team at Advanced Micro Devices, Inc. Our mission is to build great products that accelerate next-generation computing experiences.The ideal candidate will have a strong background in computer engineering and experience in silicon design delivery. They will be responsible for driving...


  • San Jose, California, United States Advanced Micro Devices , Inc. Full time

    Transforming Lives with AMD TechnologyWe're passionate about pushing the limits of innovation to solve the world's most important challenges. As a Senior Software Development Engineer at AMD, you'll join a team of talented engineers who are dedicated to building great products that accelerate next-generation computing experiences.The Role:In this role,...


  • San Diego, California, United States Advanced Micro Devices , Inc. Full time

    Transform Lives with AMD TechnologyWe are dedicated to enriching our industry, communities, and the world through innovative technology solutions. Our mission is to accelerate next-generation computing experiences, driving progress in the data center, artificial intelligence, PCs, gaming, and embedded sectors. At AMD, we foster a culture of innovation,...


  • San Jose, United States Advanced Micro Devices , Inc. Full time

    WHAT 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, United States AMD Full time

    WHAT 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...

Machine Learning Compiler Software Engineer for Advanced Micro Devices, Inc.

2 months ago


San Jose California, United States Advanced Micro Devices, Inc Full time
About the Role

We are seeking a highly skilled Machine Learning (ML) Compiler Software Engineer to join our team at Advanced Micro Devices, Inc. (AMD) in the AI group. 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 Neural Processing Units (NPUs).

Responsibilities
  • Design and implement compiler optimization passes that translate Gen-AI ML inference models into low-level code for compute and dataflow on NPUs.
  • Analyze and transform intermediate representations of ML models (computational graphs) for efficient execution.
  • 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.
Requirements
  • 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.
Qualifications

At AMD, your base pay is one part of your total rewards package. Your base pay will depend on where your skills, qualifications, experience, and location fit into the hiring range for the position. You may be eligible for incentives based upon your role such as either an annual bonus or sales incentive. Many AMD employees have the opportunity to own shares of AMD stock, as well as a discount when purchasing AMD stock if voluntarily participating in AMD's Employee Stock Purchase Plan. You'll also be eligible for competitive benefits described in more detail.