
About Amazon
The everything store and cloud computing leader
Key Highlights
- Headquartered in South Lake Union, Seattle, WA
- Over 1.5 million employees worldwide
- Leading cloud services through Amazon Web Services (AWS)
- Acquired Whole Foods, Twitch, and Ring
Amazon, headquartered in South Lake Union, Seattle, WA, is the world's largest online retailer and a leader in cloud computing through Amazon Web Services (AWS). With over 1.5 million employees globally, Amazon operates in various sectors, including AI with its Alexa devices and a vast marketplace k...
🎁 Benefits
Amazon offers competitive salaries, stock options, generous PTO policies, and comprehensive health benefits. Employees also have access to a learning ...
🌟 Culture
Amazon's culture is driven by customer obsession and a focus on innovation. The company encourages employees to think big and move fast, fostering an ...
Skills & Technologies
Overview
Amazon is hiring a Software Development Manager for the Neuron Tools team to lead the development of high-performance monitoring and profiling tools for machine learning applications. You'll work with C++, PyTorch, and TensorFlow to optimize AI workloads. This position requires a strong background in AI/ML tools and performance analysis.
Job Description
Who you are
You have a proven track record of leading engineering teams and developing high-performance tools for machine learning applications. With extensive experience in C++, you understand the intricacies of performance optimization and profiling in AI workloads. Your background includes building AI/ML tools, and you are familiar with frameworks like PyTorch and TensorFlow, which are essential for this role. You possess strong leadership skills and are motivated to drive results while collaborating with cross-functional teams.
Your experience includes direct customer-facing roles, allowing you to understand user needs and translate them into actionable insights for tool development. You are passionate about innovation and are eager to explore new technologies that enhance the performance of machine learning applications. You thrive in a fast-paced environment and are committed to delivering high-quality solutions that meet customer expectations.
Desirable
Experience with ML-specific profiler tools, such as PyTorch Profiler or TensorFlow Profiler, is highly desirable. Familiarity with cloud-scale machine learning accelerators and the ability to support multiple frameworks will set you apart as a candidate. You are also open to mentoring junior engineers and fostering a collaborative team culture.
What you'll do
As the Software Development Manager for the Tools Team, you will oversee the design, development, and deployment of the Neuron Profiler and other Neuron Tools. Your role involves managing the full development life cycle of these tools, ensuring they are scalable, reliable, and user-friendly. You will collaborate with cross-functional teams to ensure that the C++ compiler and runtime generate key performance insights for customers.
You will lead a talented team of engineers, guiding them in developing high-performance monitoring and profiling tools that optimize AI workloads across various hardware platforms. Your responsibilities will include driving innovations that allow the profiler to support multiple frameworks, enhancing the overall user experience. You will also be responsible for ensuring that the tools meet the needs of both internal and external customers, providing deep insights into performance bottlenecks and system behavior.
What we offer
At Amazon, we offer a competitive salary range of $184,900.00 - $250,200.00 USD annually for this position. You will have access to a comprehensive benefits package, including health insurance, retirement plans, and paid time off. Our work environment encourages innovation and collaboration, allowing you to make a significant impact in the field of machine learning and AI. Join us in shaping the future of cloud-scale machine learning accelerators and tools.
Interested in this role?
Apply now or save it for later. Get alerts for similar jobs at Amazon.
Similar Jobs You Might Like
Based on your interests and this role

Software Engineering Manager
Amazon is seeking a Software Engineering Manager for their ML Accelerators team at AWS Neuron. You'll lead a team of compiler engineers to design and implement machine learning compiler solutions using technologies like Python, TensorFlow, and PyTorch. This role requires strong leadership and technical expertise in machine learning.

Software Development Engineer
Amazon is hiring a Software Development Engineer for the Neuron Foundation Tools Team to develop high-performance monitoring and profiling tools for machine learning applications. You'll work with C++, AWS, and various ML frameworks. This position requires experience in building AI/ML tools.

Software Development Engineer
Amazon is hiring a Senior Software Development Engineer to lead a technical team focused on profiling and optimization tools for ML accelerators. You'll work with AWS and Python to enhance performance for large ML workloads. This position requires strong experience in machine learning and software development.

Software Development Engineer
Amazon is hiring a Software Development Engineer for the Neuron Foundation Tools Team to develop high-performance monitoring and profiling tools for machine learning applications. You'll work with C++, AWS, and various ML frameworks. This position requires experience in building AI/ML tools.

Software Development Manager
Amazon is hiring a Software Development Manager for AI Inference Technology to lead a team optimizing large language models for Trainium. You'll work with AWS, PyTorch, and distributed systems. This position requires a strong background in machine learning and team management.