- Amazon (Seattle, WA)
- …and Trainium ML accelerators. This comprehensive toolkit includes an ML compiler , runtime, and application framework that seamlessly integrates with ... popular ML frameworks like PyTorch and JAX enabling unparalleled ...layers - from frameworks and kernels and collaborate with compiler to runtime and collectives. We not only optimize… more
- Amazon (Seattle, WA)
- …the Trn2 and future Trn3 servers that use them. This role is for a software engineer in the Machine Learning Applications ( ML Apps) team for AWS Neuron. This ... enables and performance tunes building blocks for all key ML model families, including Llama3, GPT OSS, Qwen3, DeepSeek...works side by side with the Inference Model Enablement, compiler runtime engineers to create, build and tune high-performance… more
- Amazon (Seattle, WA)
- …and Trainium cloud-scale machine learning accelerators. This role is for a senior software engineer in the Machine Learning Inference Applications team. ... of Experts, etc. The team works side by side with chip architects, compiler engineers and runtime engineers to deliver performance and accuracy on Neuron devices… more
- Amazon (Seattle, WA)
- …In this role, you will drive the design and implementation of ML compiler solutions, develop advanced optimization techniques, lead hardware bring-up ... stack, the AWS Neuron Software Development Kit, which includes a Machine Learning compiler . The SDK natively integrates into popular ML frameworks, such as… more
- Amazon (Seattle, WA)
- …and Trainium cloud-scale machine learning accelerators. This role is for a senior software engineer in the Machine Learning Inference Applications team. ... of Experts, etc. The team works side by side with chip architects, compiler engineers and runtime engineers to deliver performance and accuracy on Neuron devices… 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 ... ML Apps team works side by side with compiler engineers and runtime engineers to create, build and...building an environment that celebrates knowledge-sharing and mentorship. Our senior members enjoy one-on-one mentoring and thorough, but kind,… more
- Amazon (Seattle, WA)
- …and the Trn1 and Inf1 servers that use them. As the Software Development Engineer for the Neuron Foundation Tools Team, you will be responsible for working alongside ... into performance bottlenecks and system behavior. Improving performance of ML Kernels and ML Frameworks. In this...with cross-functional teams to ensure that the our C++ compiler and runtime generates key information so customers can… more