Software Development Manager, Neuron Tools, Annapurna Labs jobs in United States
cer-icon
Apply on Employer Site
company-logo

Amazon · 1 day ago

Software Development Manager, Neuron Tools, Annapurna Labs

Amazon is a leading technology company, and they are seeking a Software Development Manager for the Neuron Tools Team. The role involves leading a team to develop and maintain high-performance monitoring and profiling tools for machine learning applications and AI accelerators, ensuring scalability, reliability, and usability of the toolchain.

Artificial Intelligence (AI)DeliveryE-CommerceFoundational AIRetail
check
H1B Sponsor Likelynote

Responsibilities

Lead a talented team of engineers to develop and maintain high-performance monitoring and profiling tools for machine learning applications and AI accelerators
Oversee the design, development, and deployment of the Neuron Profiler and other Neuron Tools
Manage the full development life cycle of the Neuron Profiler/Tools toolchain
Collaborate with cross-functional teams to ensure that the C++ compiler and runtime generates key information for customers
Drive innovations that allow the profiler to support multiple frameworks, such as PyTorch, TensorFlow, and XLA
Work with the executive leadership and other senior management and technical leaders to define product directions and deliver them to customers

Qualification

C++Machine LearningPerformance ProfilingPythonGoLinuxCustomer-facing ExperienceTeam ManagementCollaborationMentorship

Required

3+ years of engineering team management experience
7+ years of working directly within engineering teams experience
3+ years of designing or architecting (design patterns, reliability and scaling) of new and existing systems experience
Experience partnering with product or program management teams
Experience in C++, Go, and Python

Preferred

2+ years experience leading teams that in Machine Learning development including building and training large models, working with Pytorch and/or Tensorflow using large distributed fleets of GPU or other accelerated systems
Experience with Linux distributions such as Ubuntu or CentOS, kernel development, and tooling such as perf and gdb
Experience with performance profiling, tracing, and analysis of AI training/inference applications
Experience with large scale, distributed AI training/inference applications, including libfabric, MPI, slurm, and EKS
Experience with fleet monitoring, debugging, and reliability
Knowledge of AI-powered optimization suggestions for profiling would be an advantage for this position

Benefits

Health insurance (medical, dental, vision, prescription, Basic Life & AD&D insurance and option for Supplemental life plans, EAP, Mental Health Support, Medical Advice Line, Flexible Spending Accounts, Adoption and Surrogacy Reimbursement coverage)
401(k) matching
Paid time off
Parental leave

Company

Amazon is a tech firm with a focus on e-commerce, cloud computing, digital streaming, and artificial intelligence.

H1B Sponsorship

Amazon has a track record of offering H1B sponsorships. Please note that this does not guarantee sponsorship for this specific role. Below presents additional info for your reference. (Data Powered by US Department of Labor)
Distribution of Different Job Fields Receiving Sponsorship
Represents job field similar to this job
Trends of Total Sponsorships
2025 (22803)
2024 (21175)
2023 (19057)
2022 (24088)
2021 (12233)
2020 (14881)

Funding

Current Stage
Public Company
Total Funding
$8.11B
Key Investors
AmazonKleiner Perkins
2023-01-03Post Ipo Debt· $8B
2001-07-24Post Ipo Equity· $100M
1997-05-15IPO

Leadership Team

leader-logo
Douglas J. Herrington
CEO, Worldwide Amazon Stores
linkedin
leader-logo
Werner Vogels
VP & CTO
linkedin
Company data provided by crunchbase