Machine Learning Performance Manager

Oversee team focused on machine learning compiler tooling as a Manager III. Champion high-performance development for AWS Inferentia and Trainium, driving scaling and optimization initiatives.

You will lead compiler engineers while working directly on the Neuron SDK, aiming for industry-leading performance. Your insights will drive enhancements on pre-silicon projects and ensure system scalability and cost efficiency for complex ML workloads. Foster innovation and technical excellence within your team.

Key Responsibilities:
• Manage the team developing the Neuron SDK compiler
• Ensure high performance in ML tooling and frameworks
• Collaborate with cross-functional ML teams
• Facilitate deployment and scaling of compiler products
• Lead optimization of code generation strategies

Requirements:
• 3+ years in engineering team management
• 6+ years working within engineering projects
• 4+ years of produc...