• 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, ... and ML models including model/data sharding, mesh sizes, quantization, sparsity, compiler tuning, and other model compression techniques, new ML model… more
    Google (06/21/25)
    - Related Jobs
  • Software Engineer- AI/ML, AWS Neuron

    Amazon (Seattle, WA)
    …The Distributed CoreTech training team works side by side with chip architects, compiler engineers and runtime engineers to create , build and tune distributed ... building distributed training support into Pytorch and Jax using XLA and the Neuron compiler and runtime stacks. This role will help tune these models to ensure… more
    Amazon (06/15/25)
    - Related Jobs
  • Software Engineer II - AI/ML, AWS Neuron

    Amazon (Seattle, WA)
    …many more. The ML Apps team works side by side with chip architects, compiler engineers and runtime engineers to create , build and optimize performance and accuracy ... in building distributed inference support into PyTorch using XLA and the Neuron compiler and runtime stacks. Strong software development skills in C++/Python and ML… more
    Amazon (06/14/25)
    - Related Jobs
  • Machine Learning Engineer, AWS Neuron Apps, AWS…

    Amazon (Seattle, WA)
    …and many more. The ML Apps team works side by side with compiler engineers and runtime engineers to create, build and tune distributed inference solutions ... building distributed inference support into Pytorch, Tensorflow using XLA and the Neuron compiler and runtime stacks. This role will help tune these models to ensure… more
    Amazon (06/14/25)
    - Related Jobs
  • Sr. Software Engineer- AI/ML, AWS Neuron Apps

    Amazon (Seattle, WA)
    …many more. The ML Apps team works side by side with chip architects, compiler engineers and runtime engineers to create , build and tune distributed training ... building distributed inference support into Pytorch, Tensorflow using XLA and the Neuron compiler and runtime stacks. This role will help tune these models to ensure… more
    Amazon (06/06/25)
    - Related Jobs
  • Sr. Software Engineer- AI/ML, AWS Neuron…

    Amazon (Seattle, WA)
    …many more. The ML Apps team works side by side with chip architects, compiler engineers and runtime engineers to create , build and tune distributed training ... training and inference support into Pytorch, Tensorflow, Jax using XLA and the Neuron compiler and runtime stacks. This role will help tune these models to ensure… more
    Amazon (05/23/25)
    - Related Jobs
  • Sr. Software Engineer- AI/ML, AWS Neuron…

    Amazon (Seattle, WA)
    …The ML Distributed Training team works side by side with chip architects, compiler engineers and runtime engineers to create, build and tune distributed training ... building distributed training support into Pytorch, Tensorflow using XLA and the Neuron compiler and runtime stacks. You will help tune these models to ensure… more
    Amazon (05/16/25)
    - Related Jobs
  • Sr. Software Engineer- AI/ML, AWS Neuron Apps

    Amazon (Seattle, WA)
    …many more. The ML Apps team works side by side with chip architects, compiler engineers and runtime engineers to create , build and tune distributed training ... training and inference support into Pytorch, Tensorflow using XLA and the Neuron compiler and runtime stacks. This role will help tune these models to ensure… more
    Amazon (04/21/25)
    - Related Jobs
  • [PhD] ML Infrastructure Engineer - Distributed…

    Amazon (Seattle, WA)
    …The ML Distributed Training team works side by side with chip architects, compiler engineers and runtime engineers to create, build and tune distributed training ... learning frameworks (PyTorch and JAX) using AWS's specialized AI hardware. Working with our compiler and runtime teams, you'll learn how to optimize ML models to run… more
    Amazon (04/16/25)
    - Related Jobs
  • Sr Software Development Engineer, AWS Open Data…

    Amazon (Redmond, WA)
    …query runtime and storage connectors. We have also made significant changes to the compiler to enable enterprise features like fine grain access control. This is a ... "must-win" strategic area in a growing and very technical space. We are looking for a hands-on engineer to help us improve the engine further. We are looking to make deeper changes all around the query engine codebase. As a Sr Engineer on the engines team, you… more
    Amazon (06/25/25)
    - Related Jobs