- 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...to build distributed training support into PyTorch, the Neuron compiler , and runtime stacks. You will enable distribute training… more
- 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 (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 (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 (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 (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
- Microsoft Corporation (Redmond, WA)
- …across Azure and Microsoft products. We collaborate closely with hardware architects, compiler teams, and model experts to co-design software solutions that unlock ... stack targeting Microsoft's 1P AI accelerators. + Collaborate with hardware, compiler , and model teams to develop high-performance solutions. + Contribute to… more
- Google (Seattle, WA)
- Software Engineer , PhD, Early Career, AI/Machine Learning, 2026 Start _corporate_fare_ Google _place_ Sunnyvale, CA, USA; Atlanta, GA, USA; +9 more; +8 more **Mid** ... on and is growing every day. As a software engineer , you will work on a specific project critical...across the full stack, from low-level hardware acceleration and compiler optimizations to high-level model architecture and production APIs,… more
- Amazon (Seattle, WA)
- …AWS's next-generation AI accelerators Inferentia and Trainium. As a Senior Software Engineer in our Machine Learning Applications team, you'll be at the forefront ... and vision generative AI models * Collaborate directly with silicon architects and compiler teams to push the boundaries of AI acceleration * Drive performance… more
- Amazon (Seattle, WA)
- …designed by Annapurna Labs inside AWS. The Neuron SDK consists of a compiler , runtime, frameworks, and tooling customers need. It's also preinstalled in AWS Deep ... inference and training. This position is for a Software Engineer for the AWS Neuron SDK team with a...from software to hardware, is also a necessity. This engineer will work with the team to define the… more