- Amazon (Seattle, WA)
- …our custom chips designed to accelerate deep-learning workloads. This role is for a software engineer in the Compiler team for AWS Neuron. As part of this role, ... you will be responsible for building next generation Neuron compiler which transforms ML models written in ML frameworks (eg, PyTorch, TensorFlow, and JAX) to be… 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 ... and create high-performance kernels for ML functions, ensuring every compute unit is fine tuned for optimal performance for...layers - from frameworks and kernels and collaborate with compiler to runtime and collectives. We not only optimize… 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 ... and create high-performance kernels for ML functions, ensuring every compute unit is fine tuned for optimal performance for...layers - from frameworks and kernels and collaborate with compiler to runtime and collectives. We not only optimize… 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...develop necessary tools to increase efficiency in large scale compute . About the team Our team is dedicated to… more