Software Development Manager, AWS Neuron
$93k - $128kAnnapurna Labs (U.S.) Inc.
Salary: $93,000 - 128,000 per year Requirements:
- We require at least 2 years of engineering team management experience.
- We look for 6+ years of direct experience working within engineering teams.
- We need 4+ years of experience designing or architecting new and existing systems, including design patterns, reliability, and scaling.
- We expect experience partnering with product or program management teams.
- We require a solid understanding of compilers, including resource management, instruction scheduling, code generation, and compute graph optimization.
- We need strong software design fundamentals and excellent system-level coding skills.
- A masters degree or Ph.D. in Computer Science or a related technical field is preferred.
- Candidates must currently be located in, or willing to relocate to, Cupertino, CA.
- We will have you lead, develop, and mentor a team of compiler engineers, including hiring, career progression, and team health.
- You will own the execution and delivery of NKI compiler features across release cycles, balancing scope, quality, and timelines.
- You will set the technical direction and roadmap for the NKI compiler area in collaboration with senior engineers.
- You will partner across framework, kernel, runtime, hardware pre-silicon design, and product management teams to launch new capabilities.
- You will drive resolution of complex, high-priority technical issues while staying close enough to the code and hardware to guide the team.
- You will represent your teams work and priorities to senior leadership and cross-organizational stakeholders.
- You will anchor decisions in customer impact, ensuring we build what unblocks real ML workloads on Trainium.
- You will support roadmap planning, technical reviews, profiling analysis, and escalation handling to keep the team moving quickly.
- AWS
- Cloud
- Hardware
- Support
- Machine Learning
- Flow
- PyTorch
More:
We are Amazon Annapurna Labs, building innovative silicon and software for AWS customers at cloud scale. Our AWS Neuron and Trainium products power large-scale machine learning workloads through the Neuron Software Development Kit, including the Neuron Kernel Interface for directly programming NeuronDevices on AWS Trainium instances. We serve customers such as Epic Games, Snap, Airbnb, Autodesk, Amazon Alexa, and Amazon Rekognition, among many others. Our team is small, agile, and highly collaborative, bringing together experts in silicon engineering, hardware design and verification, software, and operations. We value inclusion, continuous learning, work-life balance, mentorship, and career growth, and this role is based in Cupertino, CA with comprehensive benefits including health coverage, sign-on payments, and RSUs.
last updated 33 week of 2026
$212.7k - $287.7k
AWS Neuron is the complete software stack for the AWS Inferentia and Trainium cloud-scale machine learning... ...use them. As the SDM of Software Development for the Neuron Training team, you... ...leading a strong team of engineers and managers to help design and deploy these new...Amazon Web ServiceLocal areaWork from homeFlexible hours$212.7k - $287.7k
The Product: AWS Machine Learning accelerators are at the forefront of AWS innovation... ...cloud. This is all enabled by the AWS Neuron Software Development Kit (SDK), which includes an ML... ...seeking a talented Software Development Manager with strong leadership and mentoring skills...Amazon Web ServiceLocal areaWork from homeRelocationFlexible hoursDay shift$212.7k - $287.7k
We develop AWS Neuron, the complete software stack for Trainium, Amazon's custom cloud-scalemachine learning... .... You should be capable of managing demanding, fast-changing priorities... ...engineering team* Report on status of development, quality, operations, and model performance...Amazon Web ServiceLocal areaFlexible hours$212.7k
...working with PyTorch or JAX software. We need 3+ years of engineering team management experience. We need... ...in production on GPUs, Neuron, TPU, or other AI... ...Technologies: AI AWS Cloud CUDA Hardware... ...working practices, and development opportunities. The role...Amazon Web ServiceFull timeFlexible hours$212.7k - $287.7k
The Product: AWS Machine Learning accelerators are at the forefront of AWS... .... This is all enabled by edge software stack, the AWS Neuron Software Development Kit (SDK), which includes an ML compiler... ...seeking a talented SW Engineering Manager with strong leadership/ mentoring...Amazon Web ServiceLocal areaWork from homeRelocationFlexible hours$212.7k
...5 years of engineering team management experience. We need 9+ years... .... We require strong software design fundamentals and excellent... ...LLMs in production on GPUs, Neuron, TPU, or other AI acceleration... ...new compiler for our custom AWS hardware at scale. We will...Amazon Web ServiceFull time$212.7k - $287.7k
How do you help every AWS customer understand who can access what, and why, across billions of resource configurations? That's the challenge at the core of this role.As a Software Development Manager on the AWS IAM Access Analyzer team, you'll lead a team building the next...Amazon Web ServiceLocal areaFlexible hours$212.7k - $287.7k
...always available at massive scale. HyperPlane is looking for a Software Development Manager (SDM III) to grow a team of 14 engineers solving never-... ...Virtualization (NFV) platform. Networking products that millions of AWS customers rely on run on top of HyperPlane, including...Amazon Web ServiceLocal areaFlexible hoursShift work$212.7k - $287.7k
...kinds of challenges? Are you skilled at managing complex projects that require... ...is part of the NPD (Network Product Development) team. As a leader in this team, you... ...communities, network engineering, peer software teams and other AWS service teams to develop and operate...Amazon Web ServiceLocal areaFlexible hours$165.2k - $223.6k
...was a startup acquired by AWS in 2015 and is now fully integrated. AWS Neuron is the complete software stack for the Inferentia and... ...is looking for a Software Development Engineer to build platform... ...for ML accelerator resource management Build and maintain Neuron...Amazon Web ServiceInternshipLocal areaFlexible hours$193.3k - $261.5k
Annapurna Labs is an integral part of AWS and develops hardware and software components that are critical... ...AWS customer experience. The AWS Neuron Collectives team is seeking a Software... ...exercise sound judgment, effectively manage stress and work safely and respectfully...Amazon Web ServiceLocal areaWork from homeFlexible hours$165.2k - $223.6k
...AWS Neuron is the complete software stack for the AWS Inferentia and Trainium cloud-scale machine learning... ...that use them. As the Software Development Engineer for the Neuron Runtime Team... ...Frameworks. In this role, you will manage the full development life cycle of...Amazon Web ServiceInternshipLocal areaWork from homeFlexible hours$193.3k - $261.5k
...be part of AI revolution? At AWS our vision is to make deep... ...vision, we’ve created innovative software and hardware solutions that make it possible. AWS Neuron is the SDK that optimizes... ..., you’ll support the development and management of Compute, Database, Storage...Amazon Web ServiceLocal areaFlexible hours$193.5k
...~5+ years of experience building complex software systems that have been successfully delivered... ...to maximize training performance Use Neuron Explorer and similar tools to pinpoint... ...automation solutions Technologies: AI AWS EC2 Firmware Hardware Support...Amazon Web ServiceFull timeFlexible hours$143.4k - $165.6k
...need 3+ years of professional software development experience outside of... ...production experience with AWS services such as EC2, ECS, CloudWatch... ..., and deployment of Neuron Runtime and related Neuron components... ...and ML frameworks. We manage the full development...Amazon Web ServiceFull timeInternshipFlexible hours$193.3k - $261.5k
We develop AWS Neuron, the complete software stack for Trainium, Amazon's custom cloud-scale machine learning... ...hardware.As a Sr. Software Development Engineer on the Inference Model Enablement... ..., system engineers, and product managers to architect and deliver state-of-...Amazon Web ServiceInternshipLocal areaFlexible hours$193.3k - $261.5k
...experience building complex software systems that have been successfully... .... We use tools such as Neuron Explorer to pinpoint... ...solutions. Technologies: AI AWS EC2 Firmware Hardware... ...balance, mentorship, career development, and comprehensive benefits,...Amazon Web ServiceFull time$193.3k - $261.5k
The Annapurna Labs team at Amazon Web Services (AWS) builds AWS Neuron, the software development kit used to accelerate deep learning and GenAI workloads on... ...of applied scientists, system engineers, and product managers to deliver state-of-the-art inference capabilities...Amazon Web ServiceInternshipLocal areaFlexible hours$193.5k
...prefer experience with resource management, scheduling, code generation,... ..., test, deploy, and maintain software that improves the performance... ..., and user experience of the Neuron compiler. We develop the... ..., and JAX for deployment on AWS Inferentia and Trainium servers...Amazon Web ServiceFull time$165.2k - $223.6k
As a Neuron Collectives Software Developer, you will:* Enhance collective algorithms... ...low level device driver development* Work closely with the... ...lifeAnnapurna Labs, a crucial part of AWS, is responsible for... ...reviews, source control management, build processes, testing,...Amazon Web ServiceInternshipLocal areaFlexible hours$212.7k - $287.7k
The Annapurna Labs team at Amazon Web Services (AWS) builds AWS Neuron, the software development kit used to accelerate deep learning and GenAI workloads... ...Basic qualifications- 3+ years of engineering team management experience- 7+ years of working directly within engineering...Amazon Web ServiceLocal areaWork from homeFlexible hours$171.6k - $222.2k
The AWS Neuron Science Team is looking for talented scientists to enhance our software stack, accelerating customer adoption of Trainium and... ...- Efficient Kernel Development: Designing high-performance... ...sound judgment, effectively manage stress and work safely and respectfully...Amazon Web ServiceLocal areaFlexible hours$165.6k
...non-internship professional software development experience We require 2+... ...code reviews, source control management, build processes, testing, and... ...We prefer experience with AWS compute services such as EC2... ...management We build and maintain Neuron DLCs and DLAMIs for customer...Amazon Web ServiceFull timeInternship$165.2k - $223.6k
The Annapurna Labs team at Amazon Web Services (AWS) builds AWS Neuron, the software development kit used to accelerate deep learning and GenAI workloads on... ...of applied scientists, system engineers, and product managers to deliver state-of-the-art inference capabilities...Amazon Web ServiceWork experience placementInternshipLocal areaFlexible hours$165.2k - $223.6k
AWS Neuron is the complete software stack for the AWS Inferentia and Trainium cloud-scale machine learning... ...Software Engineer that will lead the development of various services that will aid... ...knowledge of optimization, resource management, scheduling are needed. The ideal...Amazon Web ServiceWork experience placementInternshipLocal areaFlexible hours$193.3k - $261.5k
The Product: AWS Machine Learning accelerators are at the forefront of... ...This is all enabled by cutting edge software stack, the AWS Neuron Software Development Kit (SDK), which includes an ML... ..., code reviews, source control management, build processes, testing, and operations...Amazon Web ServiceInternshipLocal areaWork from homeRelocationFlexible hours$171.6k - $222.2k
The AWS Neuron Science Team is looking for talented scientists to enhance our software stack, accelerating customer adoption of Trainium... ...validation- Efficient Kernel Development: Designing high-performance... ...the development and management of Compute, Database, Storage...Amazon Web ServiceLocal areaFlexible hours- ...product(s)· Products evaluated will be deployed in SaaS (likely AWS hosted by SPM vendor), so SaaS understanding is greatly... ...to be effective in the role · Experience in Sales Performance Management is a big plus, as space is quite niche and fairly complexSRI Tech...Amazon Web ServiceFull time
$212.7k - $287.7k
Amazon Web Services is looking for an experienced Software Development Manager to lead a two-pizza team building the next generation of access control for AWS Glue Data Catalog. This is a high-impact, technically complex role at the intersection of data governance, security...Amazon Web ServiceLocal areaFlexible hours$253.1k - $342.3k
Amazon Elastic MapReduce (EMR) is AWS's cloud big data platform for running large-... ...models — from fully serverless to customer-managed clusters. Organizations across... ...workloads.We are looking for a Senior Software Development Manager to lead EMR's Workload Management...Amazon Web ServiceLocal areaFlexible hours
Do you want to receive more vacancies?
Subscribe and receive similar vacancies to Software Development Manager, AWS Neuron. Be the first to apply!
- software manager Cupertino, CA
- director software engineering Cupertino, CA
- director of software Cupertino, CA
- application manager Cupertino, CA
- IT software development manager Cupertino, CA
- software asset management analyst Cupertino, CA
- ultimate software Cupertino, CA
- embedded software Cupertino, CA
- software Cupertino, CA
- software engineer - cloud services Cupertino, CA


