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


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


  • 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 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 PayPal, Inc. Full time

    Job Title: Senior Machine Learning EngineerJob Summary:PayPal, Inc. is seeking a highly skilled Senior Machine Learning Engineer to join our Global Analytics and Data Science team. As a Senior Machine Learning Engineer, you will be responsible for developing and enhancing machine learning/AI capabilities to drive business growth and innovation.Key...


  • 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 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 Tik Tok Full time

    Job Title: Senior Machine Learning EngineerAbout the Role:We are seeking a highly skilled Senior Machine Learning Engineer to join our Developer Infrastructure team at TikTok. As a key member of our team, you will be responsible for developing and deploying machine learning systems that enable our developers to build large-scale applications with...


  • San Jose, California, United States Adobe Full time

    We are seeking a highly skilled Senior Machine Learning Engineer to join our team at Adobe. The ideal candidate will have a strong background in machine learning and a passion for building innovative AI solutions.The successful candidate will be responsible for designing, developing, and deploying machine learning models and algorithms to drive business...


  • 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 AMD Full time

    Transforming Lives with AMD TechnologyWe are committed to enriching our industry, communities, and the world through innovative 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. At AMD, we push the limits...


  • San Jose, California, United States Hireio, Inc. Full time

    Job DescriptionHireio, Inc. is seeking a skilled Senior Machine Learning Engineer to join our Brand Ads Team. This team owns several ads products, including reservation ads, auction ads, and innovative content ads that enable advertisers and users to foster brand awareness and achieve business goals.We are looking for engineers with strong problem-solving...


  • San Jose, California, United States Hireio, Inc. Full time

    Job OverviewHireio, Inc. is seeking a seasoned Senior Machine Learning Engineer to join our Brand Ads Team. As a key member of our team, you will be responsible for creating innovative monetization products that drive engagement and revenue.Key Responsibilities:Create innovative monetization products that drive engagement and revenue.Participate in the...


  • San Jose, California, United States Tik Tok Full time

    Developer Infrastructure TeamWe are seeking a highly skilled Machine Learning Engineer to join our Developer Infrastructure team at TikTok. As a senior member of our team, you will play a key role in developing and maintaining our machine learning systems, including key computing development, task scheduling, and machine learning system management and...


  • San Jose, California, United States AMD Full time

    About the RoleWe are seeking a talented Machine Learning (ML) Compiler Software Engineer to join our growing team in the AI group at AMD. As a key member of our team, you will play a crucial role in developing software tools to deploy cutting-edge ML models on AMD's XDNA Neural Processing Units (NPU). Your primary responsibility will be to design, implement,...


  • San Jose, California, United States Tik Tok Full time

    Job DescriptionTikTok is a leading destination for short-form mobile video, and our mission is to inspire creativity and bring joy. We're looking for a talented Senior Machine Learning Engineer to join our Ads Identity and Attribution team, which is at the forefront of addressing the shift in online advertising.As a key member of our team, you'll have the...


  • San Jose, California, United States Tik Tok Full time

    Job Description:At TikTok, we're seeking a highly skilled Senior Machine Learning Engineer to join our Developer Infrastructure team. As a key member of our team, you will be responsible for leading the development of machine learning algorithms and systems, fine-tuning and training machine learning models, and iterating on the machine learning system using...

Senior Machine Learning Compiler Engineer

2 months ago


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

We are seeking a highly skilled and experienced Machine Learning Compiler Engineer to join our team at Advanced Micro Devices, Inc. As a key member of our core team, you will be responsible for developing a cutting-edge machine learning model compiler targeting AMD Inference Accelerator AIE hardware devices.

Key Responsibilities
  • Implement and improve passes in the compiler to optimize performance and efficiency.
  • Integrate the compiler and compiled model with popular machine learning frameworks such as PyTorch and TensorFlow.
  • Implement model partitioning in machine learning frameworks and/or MLIR to ensure seamless integration with AMD hardware.
  • Design and develop a runtime to distribute work to and collect results from x86 cores and the array of AIE cores.
  • Mentor and provide guidance to junior team members to ensure knowledge sharing and skill development.
  • Stay up-to-date with the latest industry trends and bring new ideas to the team to drive innovation.
  • Collaborate with cross-functional teams to design and develop new groundbreaking AMD technologies.
  • Debug and fix existing issues, and research alternative, more efficient ways to accomplish the same work.
  • Develop technical relationships with peers and partners to drive business growth and success.
Requirements
  • Strong object-oriented programming background, with expertise in C/C++ and Python.
  • Experience with machine learning compiler and runtime technologies, including OneDNN, MLIR, XLA, OpenXLA, IREE, and OpenAI Triton compiler.
  • Compiler building skills, with a focus on code generation for machine learning hardware accelerators.
  • Experience with GPU code generation and machine learning concepts and model development.
  • Understanding of popular machine learning frameworks, including PyTorch, TensorFlow, ONNX, and JAX.
  • Ability to write high-quality code with a keen attention to detail, and experience with modern concurrent programming and threading APIs.
  • Experience with software development processes and tools, including debuggers, source code control systems, and profilers.
  • Effective communication and problem-solving skills, with the ability to motivate and lead a team.
Preferred Qualifications
  • Bachelor's, Master's, or PhD degree in Computer Science, Computer Engineering, Electrical Engineering, or equivalent.