- Microsoft Corporation (Redmond, WA)
- Join the AI Compiler Frontend team at Microsoft, where we pioneer kernel fusion compilation techniques, develop advanced AI graph compilation tools, and design new ... Inductor. Our interdisciplinary team operates at the crossroads of compiler engineering, programming language design, algorithmic innovation, artificial intelligence,… more
- Amazon (Seattle, WA)
- …and Trainium ML accelerators. This comprehensive toolkit includes an ML compiler , runtime, and application framework that seamlessly integrates with popular ML ... layers - from frameworks and kernels and collaborate with compiler to runtime and collectives. We not only optimize...Enablement and Acceleration team works side by side with compiler engineers and runtime engineers to create, build and… more
- Amazon (Seattle, WA)
- …the Neuron Core Tools team * Work closely with the frameworks and compiler teams. * Collect requirements from various other teams including training, inference and ... runtime. * Collaborate with the compiler performance team to ensure timely release of the...AWS Neuron. Neuron is a Software that include ML compiler and native integration into popular ML frameworks. Our… more
- 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
- 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
- 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 (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 ... will lead efforts to build distributed training support into PyTorch, the Neuron compiler , and runtime stacks. You will enable distribute training strategies as well… more
- 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 ... build distributed training support into PyTorch and JAX using XLA, the Neuron compiler , and runtime stacks. You will optimize models to achieve peak performance and… more
- 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 (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