Machine Learning - Compiler Engineer II, Annapurna Labs

Amazon

Seattle, WA, US
Not specified; not specified; not specified
On-site
Deep learning compiler development
Neural network optimization
Aws neuron sdk integration
The role involves architecting and implementing business-critical features for a ground-up developed compiler handling the world's largest ML workloads

Job Summary

  • The role involves architecting and implementing business-critical features for a ground-up developed compiler handling the world's largest ML workloads.
  • Candidates will leverage technical communication skills as hands-on partners to AWS ML services teams and participate in pre-silicon design.
  • The team fosters an inclusive culture with mentorship opportunities, focusing on creating a toolchain that provides a quantum leap in performance.

Matching Summary

The role involves architecting and implementing business-critical features for a ground-up developed compiler handling the world's largest ML workloads.

Salary

Not specified; Not specified; Not specified

Skills & Requirements

Must-have

  • Deep learning compiler development
  • Neural network optimization
  • AWS Neuron SDK integration
  • PyTorch TensorFlow MxNet frameworks
  • Silicon design collaboration

Nice-to-have

  • Background in AI accelerators
  • Publishing cutting-edge research
  • Technical communication skills
  • Hands-on partnership with services teams

Key Requirements

  • Machine Learning Compiler Engineer II level experience
  • Experience with ML frameworks like PyTorch or TensorFlow
  • Background in machine learning and AI accelerators preferred

Work Rights

Not specified

Tailored Resume

Cover Letter