Software Development Manager, ML Accelerators, AWS Neuron, Annapurna Labs
Company: Amazon
Location: Seattle
Posted on: April 1, 2026
|
|
|
Job Description:
AWS Machine Learning accelerators are at the forefront of AWS
innovation. The Trainium chip delivers industry-leading ML
inference and training performance at the lowest cost in the cloud.
This is enabled by edge software stack, the AWS Neuron Software
Development Kit, which includes a Machine Learning compiler. The
SDK natively integrates into popular ML frameworks, such as
PyTorch, JAX and TensorFlow. AWS Neuron is widely adopted by many
internal and external customers and partners. Amazon Annapurna Labs
drives innovation in silicon and software for AWS, blending
cloud-scale impact with world-class engineering talent. Our
multidisciplinary team spans silicon design, hardware verification,
software, and operations. We operate in large, complex domains with
small, agile teams, fostering continuous learning and rapid
innovation. With no set blueprint, we thrive on experimentation and
offer a uniquely dynamic and enriching environment across a wide
range of AWS products and services. Learn more about our history:
https://www.amazon.science/how-silicon-innovation-became-the-secret-sauce-behind-awss-success
We are seeking an exceptional Software Engineering Manager to lead
portion of our Deep-Learning Backend Compiler team at AWS Neuron.
You will manage a team of talented compiler engineers focused on
machine learning compiler design and development. In this role, you
will drive the design and implementation of ML compiler solutions,
develop advanced optimization techniques, lead hardware bring-up
for next-generation chips, and influence pre-silicon design
decisions while mentoring your team. This position offers the
opportunity to directly impact AWS's machine learning
infrastructure and accelerate the delivery of innovative products
and features to market. Explore the Product:
https://awsdocs-neuron.readthedocs-hosted.com/en/latest/neuron-guide/neuron-cc/index.html,
https://github.com/aws/aws-neuron-sdk
https://aws.amazon.com/machine-learning/neuron/,
https://aws.amazon.com/machine-learning/neuron/ A day in the life
About the team Inclusive Team Culture Here at Annapurna Labs, we
embrace our differences. We are committed to furthering our culture
of inclusion. We have ten employee-led affinity groups, reaching
40,000 employees in over 190 chapters globally. We have innovative
benefit offerings, and host annual and ongoing learning
experiences, including our Conversations on Race and Ethnicity
(CORE) and AmazeCon conferences. Amazon’s culture of inclusion is
reinforced within our 14 Leadership Principles, which remind team
members to seek diverse perspectives, learn and be curious, and
earn trust. Work/Life Balance Our team puts a high value on
work-life balance. It isn’t about how many hours you spend at home
or at work; it’s about the flow you establish that brings energy to
both parts of your life. We believe striking the right balance
between your personal and professional life is critical to
life-long happiness and fulfillment. We offer flexibility in
working hours and encourage you to find your own balance between
your work and personal lives. Mentorship & Career Growth Our team
is dedicated to supporting new members. We have a broad mix of
experience levels and tenures, and we’re building an environment
that celebrates knowledge sharing and mentorship. Our senior
members enjoy one-on-one mentoring. We care about your career
growth and strive to assign projects based on what will help each
team member develop into a better-rounded engineer and enable them
to take on more complex tasks in the future. - Bachelor's degree or
foreign equivalent in Computer Science, Engineering, Mathematics,
or a related field - 5 years of Software Engineer, Software
Developer, or related occupational experience - 3 years of
engineering team management experience - Proven track record in new
hardware bring-up - Solid Knowledge of computer/chip architecture
and code generation - Experience with AI/ML technologies -
Experience in compiler design and architecture - Knowledge of LLVM
and MLIR Amazon is an equal opportunity employer and does not
discriminate on the basis of protected veteran status, disability,
or other legally protected status. Our inclusive culture empowers
Amazonians to deliver the best results for our customers. If you
have a disability and need a workplace accommodation or adjustment
during the application and hiring process, including support for
the interview or onboarding process, please visit
https://amazon.jobs/content/en/how-we-hire/accommodations for more
information. If the country/region you’re applying in isn’t listed,
please contact your Recruiting Partner. Our compensation reflects
the cost of labor across several US geographic markets. The base
pay for this position ranges from $166,400/year in our lowest
geographic market up to $287,700/year in our highest geographic
market. Pay is based on a number of factors including market
location and may vary depending on job-related knowledge, skills,
and experience. Amazon is a total compensation company. Dependent
on the position offered, equity, sign-on payments, and other forms
of compensation may be provided as part of a total compensation
package, in addition to a full range of medical, financial, and/or
other benefits. For more information, please visit
https://www.aboutamazon.com/workplace/employee-benefits . This
position will remain posted until filled. Applicants should apply
via our internal or external career site.
Keywords: Amazon, Seattle , Software Development Manager, ML Accelerators, AWS Neuron, Annapurna Labs, IT / Software / Systems , Seattle, Washington