Senior ML Profiler Tools Developer - Neuron SDK

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

Benefits offered by this job

RSUs

Job summary

AWS Neuron team is seeking a Senior Software Development Engineer to develop and maintain high-performance monitoring and profiling tools for ML workloads on Inf1 and Trainium accelerators. You will own design, development, and deployment of Neuron Tools and Profiler, enabling customers to optimize performance across hardware platforms.

You will collaborate with cross-functional teams to ensure the compiler and runtime generate key performance information, driving scalable, reliable, and usable

Qualifications

• 5+ years of professional software development experience.
• 5+ years programming in at least one language.
• 5+ years leading design/architecture of systems.
• Experience mentoring or leading an engineering team.

Responsibilities

• Develop and maintain high-performance monitoring and profiling tools for ML workloads.
• Lead design, development, and deployment of the Neuron Tools (Profiler).
• Collaborate with cross-functional teams to optimize performance across Trainium and Inferentia devices.

Skills

5+ years of non-internship software de

5+ years programming experience

5+ years leading design/architecture

Mentor / tech lead experience

Education

Bachelor's degree in computer science or equivalent

Job description

AWS Neuron team is seeking a Senior Software Development Engineer to develop and maintain high-performance monitoring and profiling tools for ML workloads on Inf1 and Trainium accelerators. You will own design, development, and deployment of Neuron Tools and Profiler, enabling customers to optimize performance across hardware platforms.

You will collaborate with cross-functional teams to ensure the compiler and runtime generate key performance information, driving scalable, reliable, and usable