Talent.com
AI/ML Sr. Compiler Development Engineer
AI/ML Sr. Compiler Development EngineerAdvanced Micro Devices, Inc. • San Jose, CA, United States
AI / ML Sr. Compiler Development Engineer

AI / ML Sr. Compiler Development Engineer

Advanced Micro Devices, Inc. • San Jose, CA, United States
30+ days ago
Job type
  • Full-time
Job description

WHAT YOU DO AT AMD CHANGES EVERYTHING

At AMD, our mission is to build great products that accelerate next-generation computing experiences-from AI and data centers, to PCs, gaming and embedded systems. Grounded in a culture of innovation and collaboration, we believe real progress comes from bold ideas, human ingenuity and a shared passion to create something extraordinary. When you join AMD, you'll discover the real differentiator is our culture. We push the limits of innovation to solve the world's most important challenges-striving for execution excellence, while being direct, humble, collaborative, and inclusive of diverse perspectives. Join us as we shape the future of AI and beyond. Together, we advance your career.

THE ROLE :

We are looking for a dynamic, energetic Sr. Compiler Development Engineer to join our growing team in AI Group. In this role, the individual will be responsible architecting and defining AI workload models, dataflow, defining block level and system level performance of Neural Processing Unit (NPU), NPU network performance modeling, and performance bottleneck analysis on pre / post silicon platforms. As a member of our dynamic team, you will have the opportunity to shape the future of AI model development

THE PERSON :

  • We are looking for a candidate who possesses strong engineering skills to tackle complex challenges on AI model development work.
  • You should have experience in optimizing and accelerating CNN / Generative AI models. Person needs excellent cross team collaboration skills to succeed in this role.
  • Strong experience in developing ML compiler for efficient network mapping on NPU
  • Work with cross-functional teams to optimize various parts of the SW stack - AI Compiler, AI frameworks, device drivers, and firmware.
  • Bring up emerging ML models based on CNN, transformers and characterize performance.
  • Work on quantization, sparsity, and architecture search methods to optimize and enhance the performance, efficiency, and accuracy of Generative AI models.
  • Collaborate closely with software engineers, data scientists, and researchers to integrate AI models into software applications and platforms.

KEY RESPONSIBILITIES :

  • Research, design, and implement novel methods for efficient CNN, GEN AI models.
  • Model optimization method design including quantization, sparsity, NAS, etc.
  • Collaborate with other team members and teams.
  • Collaborate with compiler team to develop optimization strategies for the compiler.
  • PREFERRED EXPERIENCE :

  • Experience with deep learning framework, e.g., Pytorch / ONNX / TensorFlow.
  • Experience on model compression, quantization, and end-to-end inference optimization.
  • Strong coding skills in C / C++, Python required.
  • Experience with any of the following also a plus : LLMs, stable diffusion, NeRF, or text-to-video generation.
  • Solid knowledge of AI and ML concepts and techniques. Practical experience applying these concepts to solve real-world problems in the context of research or work experience.
  • Understanding the performance implications on AI acceleration of different compute, memory, and communication configurations and hardware and software implementation choices.
  • Developing and optimizing code for VLIW processors. Analyzing code for high performance CONV, GEMM and non-linear operators Deep understanding of AI frameworks, preferably ONNX.
  • ACADEMIC CREDENTIALS :

  • Minimum of a BS degree, MS or above preferred.
  • LOCATION :

    San Jose, CA (Hybrid)

    #LI-MV1

    #LI-HYBRID

    Benefits offered are described : AMD benefits at a glance.

    AMD does not accept unsolicited resumes from headhunters, recruitment agencies, or fee-based recruitment services. AMD and its subsidiaries are equal opportunity, inclusive employers 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. We encourage applications from all qualified candidates and will accommodate applicants' needs under the respective laws throughout all stages of the recruitment and selection process.

    Create a job alert for this search

    Development Engineer • San Jose, CA, United States

    Related jobs
    Sr. Machine Learning - Compiler Engineer III, AWS Neuron, Annapurna Labs

    Sr. Machine Learning - Compiler Engineer III, AWS Neuron, Annapurna Labs

    Amazon • Cupertino, CA, United States
    Full-time
    The Product : AWS Machine Learning accelerators are at the forefront of AWS innovation and one of several AWS tools used for building Generative AI on AWS. The Inferentia chip delivers best-in-class ...Show more
    Last updated: 30+ days ago • Promoted
    Senior AI Software / Compiler Architect

    Senior AI Software / Compiler Architect

    Infineon Technologies AG • San Jose, CA, United States
    Full-time
    As a Senior AI Software / Compiler Architect, you will drive ML Compiler development effort and efficiently implement scheduling and routing of ML networks. Does this sound exciting? If so, please app...Show more
    Last updated: 30+ days ago • Promoted
    AIML - Sr. Machine Learning Engineer, World Knowledge

    AIML - Sr. Machine Learning Engineer, World Knowledge

    Apple • Cupertino, CA, United States
    Full-time
    Are you excited about Generative AI and Large Language Models and eager to apply your expertise in a fast-paced, innovative environment? Join a newly created team in AIML focused on building world-...Show more
    Last updated: 30+ days ago • Promoted
    Sr. Software & AI / ML Engineer, Rights and Pricing - Apple Services Engineering

    Sr. Software & AI / ML Engineer, Rights and Pricing - Apple Services Engineering

    Apple • Cupertino, CA, United States
    Full-time
    The Apple Services Engineering team is one of the most exciting examples of Apple’s long-held passion for combining art and technology. These are the people who power the App Store, Apple TV, Apple ...Show more
    Last updated: 30+ days ago • Promoted
    Senior Staff Machine Learning Compiler Engineer

    Senior Staff Machine Learning Compiler Engineer

    XPENG • Santa Clara, CA, United States
    Full-time
    AI and autonomous driving technologies into its vehicles, including electric vehicles (EVs), electric vertical take-off and landing (eVTOL) aircraft, and robotics. With a strong focus on intelligent...Show more
    Last updated: 30+ days ago • Promoted
    Sr. Machine Learning Engineer, Monetization Engineering

    Sr. Machine Learning Engineer, Monetization Engineering

    Pinterest • San Francisco, CA, United States
    Full-time
    Millions of people around the world come to our platform to find creative ideas, dream about new possibilities and plan for memories that will last a lifetime. At Pinterest, we're on a mission to br...Show more
    Last updated: 30+ days ago • Promoted
    Sr. AI / ML Engineer

    Sr. AI / ML Engineer

    VECTRA • San Jose, CA, United States
    Full-time
    Vectra® is the leader in AI-driven threat detection and response for hybrid and multi-cloud enterprises.The Vectra AI Platform delivers integrated signal across public cloud, SaaS, identity, and da...Show more
    Last updated: 30+ days ago • Promoted
    Sr. Machine Learning / AI Engineer

    Sr. Machine Learning / AI Engineer

    Rivian • Palo Alto, CA, United States
    Full-time
    Rivian is on a mission to keep the world adventurous forever.This goes for the emissions-free Electric Adventure Vehicles we build, and the curious, courageous souls we seek to attract.As a company...Show more
    Last updated: 30+ days ago • Promoted
    Staff Machine Learning Compiler Engineer

    Staff Machine Learning Compiler Engineer

    Rivian • Palo Alto, CA, United States
    Full-time
    Rivian is on a mission to keep the world adventurous forever.This goes for the emissions-free Electric Adventure Vehicles we build, and the curious, courageous souls we seek to attract.As a company...Show more
    Last updated: 30+ days ago • Promoted
    Sr Software Engineer, AI Compiler

    Sr Software Engineer, AI Compiler

    Tenstorrent • Santa Clara, California, United States
    Full-time +1
    Tenstorrent is leading the industry on cutting-edge AI technology, revolutionizing performance expectations, ease of use, and cost efficiency. With AI redefining the computing paradigm, solutions mu...Show more
    Last updated: 30+ days ago • Promoted
    Sr ML Compiler Engineer, Annapurna Labs

    Sr ML Compiler Engineer, Annapurna Labs

    Amazon • Cupertino, CA, United States
    Full-time
    The Product : AWS Machine Learning accelerators are at the forefront of AWS innovation and one of several AWS tools used for building Generative AI on AWS. The Inferentia chip delivers best-in-class ...Show more
    Last updated: 30+ days ago • Promoted
    Sr. Staff Machine Learning Engineer, Search

    Sr. Staff Machine Learning Engineer, Search

    Pinterest • San Francisco, CA, United States
    Full-time
    Millions of people around the world come to our platform to find creative ideas, dream about new possibilities and plan for memories that will last a lifetime. At Pinterest, we're on a mission to br...Show more
    Last updated: 30+ days ago • Promoted
    AI / ML Compiler Development Engineer

    AI / ML Compiler Development Engineer

    Advanced Micro Devices, Inc. • San Jose, CA, United States
    Full-time
    WHAT YOU DO AT AMD CHANGES EVERYTHING.At AMD, our mission is to build great products that accelerate next-generation computing experiences-from AI and data centers, to PCs, gaming and embedded syst...Show more
    Last updated: 30+ days ago • Promoted
    Director Machine Learning Engineering - AI / ML Model Compiler and Applications

    Director Machine Learning Engineering - AI / ML Model Compiler and Applications

    Advanced Micro Devices, Inc. • San Jose, CA, US
    Full-time
    Overview At AMD, our mission is to build great products that accelerate next-generation computing experiences—from AI and data centers, to PCs, gaming and embedded systems.Grounded in a culture of ...Show more
    Last updated: 2 days ago • Promoted
    Software Engineer, ML Inference Compiler & Deployment, AI Frameworks

    Software Engineer, ML Inference Compiler & Deployment, AI Frameworks

    Tesla • Palo Alto, CA, United States
    Full-time
    As a Software Engineer within our Autonomy teams, you will contribute to one of the most advanced and widely deployed AI Platforms in the world, powering Autopilot and our Humanoid Robot, Optimus.I...Show more
    Last updated: 1 day ago • Promoted
    On-Device ML Compiler Engineer (Model Compilation)

    On-Device ML Compiler Engineer (Model Compilation)

    Apple • Cupertino, CA, United States
    Full-time
    Imagine being at the forefront of an evolution where innovative AI meets the elegance of Apple silicon.The On-Device Machine Learning team transforms groundbreaking research into practical applicat...Show more
    Last updated: 5 days ago • Promoted
    Sr AI / ML Engineer - CX

    Sr AI / ML Engineer - CX

    Cisco • San Jose, CA, United States
    Full-time
    CX AI Incubation team is part of CX and is focused on identifying and building breakthrough emerging solutions to support the diverse requirements of the CX organization and our customers.Are you r...Show more
    Last updated: 2 days ago • Promoted
    Senior AI / ML Tooling Engineer

    Senior AI / ML Tooling Engineer

    General Motors • San Francisco, CA, US
    Full-time
    Role : We are looking for an ML tooling engineer to build tools to analyze and optimize distillation, training, and inference of ML models. You will develop and enhance GM's internal ML tooling for ...Show more
    Last updated: 15 days ago • Promoted