• Software Development Manager - Compiler

    Amazon (Seattle, WA)
    …stack, the AWS Neuron Software Development Kit (SDK), which includes an ML compiler and runtime and natively integrates into popular ML frameworks, such as PyTorch, ... leadership and mentoring capabilities to join our Deep Learning Compiler team. In this Manager III role, you will...will help each team member develop into a better-rounded engineer and enable them to take on more complex… more
    Amazon (04/25/25)
    - Related Jobs
  • 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/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
  • Software Engineer - AI/ML, AWS Neuron…

    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...support into PyTorch and JAX using XLA, the Neuron compiler , and runtime stacks. You will optimize models to… more
    Amazon (06/25/25)
    - Related Jobs
  • Staff Software Engineer , ML Performance

    Google (Kirkland, WA)
    …System (eg, Background Theory, TensorFlow, and other tools). + Experience in compiler optimizations or related fields. + Experience in ML frameworks (eg Jax, ... on and is growing every day. As a software engineer , you will work on a specific project critical...ML models including model/data sharding, mesh sizes, quantization, sparsity, compiler tuning, and other model compression techniques, new ML… more
    Google (06/21/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 ... training team works side by side with chip architects, compiler engineers and runtime engineers to create , build...into Pytorch and Jax using XLA and the Neuron compiler and runtime stacks. This role will help tune… more
    Amazon (06/15/25)
    - Related Jobs
  • Software Engineer II - AI/ML, AWS Neuron

    Amazon (Seattle, WA)
    …Trainium cloud-scale machine learning accelerators. This role is for a senior software engineer in the Machine Learning Applications (ML Apps) team for AWS Neuron. ... Apps team works side by side with chip architects, compiler engineers and runtime engineers to create , build...inference support into PyTorch using XLA and the Neuron compiler and runtime stacks. Strong software development skills in… more
    Amazon (06/14/25)
    - Related Jobs
  • Machine Learning Engineer , 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 ... The ML Apps team works side by side with compiler engineers and runtime engineers to create, build and...support into Pytorch, Tensorflow using XLA and the Neuron compiler and runtime stacks. This role will help tune… more
    Amazon (06/14/25)
    - Related Jobs