• Sr. TPM Neuron Compiler , Annapurna Labs

    Amazon (Seattle, WA)
    Description The AWS Neuron Compiler team is seeking an experienced and motivated Sr TPM to lead and increase efficiency and predictability of our LLM model ... to quickly and efficiently prioritize and implement the most important compiler capabilities, and deliver them together with customers, inference, and training… more
    Amazon (05/06/25)
    - Related Jobs
  • Software Engineer , Systems ML - Frameworks…

    Meta (Olympia, WA)
    …AI framework organization. MTIA Software Team has been developing a comprehensive AI Compiler strategy that delivers a highly flexible platform to train & serve new ... high performance for production environments across specialized hardware architectures. The compiler stack, DL graph optimizations, and kernel authoring for specific… more
    Meta (05/09/25)
    - Related Jobs
  • Software Engineer , Systems ML - Compilers

    Meta (Redmond, WA)
    …a software engineer to support the development of the compiler tool-chain for state-of-the-art deep learning hardware components optimized for AR/VR systems. ... You will be part of our efforts to architect, design and implement a clean slate compiler for this activity and will be part of a team that includes compiler ,… more
    Meta (03/29/25)
    - Related Jobs
  • Software Development Engineer II, Annapurna…

    Amazon (Seattle, WA)
    …a Software Development Engineer to support the development infrastructure of a compiler to enable the world's largest ML workloads to run efficiently in the ... Generative AI on AWS. The Neuron Compiler Engineering team is searching for...into code suitable for execution. As a Software Development Engineer on the team, you will be responsible for… more
    Amazon (04/24/25)
    - Related Jobs
  • ML Acceleration / Framework Engineer

    Amazon (Seattle, WA)
    …(Inf1/Inf2) our cloud-scale Machine Learning accelerators. This role is for a Machine Learning Engineer on one of our AWS Neuron teams: - The ML Distributed Training ... team works side by side with chip architects, compiler engineers and runtime engineers to create, build and...based system is key. - ML Frameworks partners with compiler , runtime, and research experts to make AWS Trainium… more
    Amazon (04/16/25)
    - Related Jobs
  • Principal Applied Scientist, Neuron ARG

    Amazon (Seattle, WA)
    Description The AWS Neuron Compiler team is actively seeking applied scientists to join our efforts in developing a state-of-the-art deep learning compiler ... AI. The Automated Reasoning Group in the AWS Neuron Compiler team works on the intersection of Artificial Intelligence...in your own right as a trusted AWS Neuron engineer , innovating on behalf of our customers. Your responsibilities… more
    Amazon (05/10/25)
    - Related Jobs
  • Sr. Software Engineer - AI/ML, AWS Neuron…

    Amazon (Seattle, WA)
    …the Trn1 and Inf1 servers that use them. This role is for a senior software engineer in the Machine Learning Applications (ML Apps) team for AWS Neuron. This role is ... Apps team works side by side with chip architects, compiler engineers and runtime engineers to create , build...into Pytorch, Tensorflow, Jax using XLA and the Neuron compiler and runtime stacks. This role will help tune… more
    Amazon (05/11/25)
    - Related Jobs
  • Software Engineer - AI/ML, AWS Neuron

    Amazon (Seattle, WA)
    …and the Trn1 and Inf1 servers that use them. This role is for a software engineer in the Machine Learning Applications (ML Apps) team for AWS Neuron. This role is ... Apps team works side by side with chip architects, compiler engineers and runtime engineers to create , build...into Pytorch, Tensorflow, Jax using XLA and the Neuron compiler and runtime stacks. This role will help tune… more
    Amazon (05/06/25)
    - Related Jobs
  • Sr. Software Engineer - AI/ML, AWS Neuron…

    Amazon (Seattle, WA)
    …the Trn1 and Inf1 servers that use them. This role is for a senior software engineer in the Machine Learning Applications (ML Apps) team for AWS Neuron. This role is ... Apps team works side by side with chip architects, compiler engineers and runtime engineers to create , build...support into Pytorch, Tensorflow using XLA and the Neuron compiler and runtime stacks. This role will help tune… more
    Amazon (04/21/25)
    - Related Jobs
  • [PhD] ML Infrastructure Engineer

    Amazon (Seattle, WA)
    …Machine Learning accelerators. This role is for a Senior Machine Learning Engineer in the Distribute Training team for AWS Neuron, responsible for development, ... Training team works side by side with chip architects, compiler engineers and runtime engineers to create, build and...JAX) using AWS's specialized AI hardware. Working with our compiler and runtime teams, you'll learn how to optimize… more
    Amazon (04/16/25)
    - Related Jobs