Lead ML Profiling & Optimization Tools Engineer

Amazon; Seattle (WA); On-site; USD 168,000 - 227,000; Full time

Benefits offered by this job

Health insurance

RSUs

401(k) matching

Paid time off

Parental leave

Job summary

Amazon is seeking a senior software leader to drive the Neuron Core Tools team, delivering profiling and optimization tools for the Neuron ML accelerators fleet. You will partner with hardware and software groups to ensure fast, reliable performance profiling for large ML workloads.

The position emphasizes mentorship, architectural leadership and delivering scalable tooling that speeds kernel development and tuning across AWS ML workloads. Uplifting career growth is supported.

Qualifications

• 5+ years of non-internship professional software development experience.
• 5+ years of programming with at least one software programming language experience
• 5+ years of leading design or architecture (design patterns, reliability and scaling) of new and existing systems experience
• 5+ years of full software development life cycle, including coding standards, code reviews, source control management, build processes, testing, and operations experience
• Experience as a mentor, tech lead or leading an engineering team

Responsibilities

• Provide technical leadership to 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 core performance tools.
• Anticipate future needs based on the product roadmap and develop necessary tools to increase efficiency in large scale compute.

Job description

Amazon is seeking a senior software leader to drive the Neuron Core Tools team, delivering profiling and optimization tools for the Neuron ML accelerators fleet. You will partner with hardware and software groups to ensure fast, reliable performance profiling for large ML workloads.

The position emphasizes mentorship, architectural leadership and delivering scalable tooling that speeds kernel development and tuning across AWS ML workloads. Uplifting career growth is supported.