Applied Scientist II, Annapurna ML

Amazon • Cupertino, CA

3d

About The Position

The AWS Neuron Science Team is looking for talented scientists to enhance our
software stack, accelerating customer adoption of Trainium and Inferentia
accelerators. In this role, you will work directly with external and internal
customers to identify key adoption barriers and optimization opportunities.
You'll collaborate closely with our engineering teams to implement innovative
solutions and engage with academic and research communities to advance
state-of-the-art ML systems. As part of a strategic growth area for AWS, you'll
work alongside distinguished engineers and scientists in an exciting and
impactful environment. We actively work on these areas:

- AI for Systems: Developing and applying ML/RL approaches for kernel/code
generation and optimization
- Machine Learning Compiler: Creating advanced compiler techniques for ML
workloads
- System Robustness: Building tools for accuracy and reliability validation
- Efficient Kernel Development: Designing high-performance kernels optimized for
our ML accelerator architectures

A day in the life
AWS Utility Computing (UC) provides product innovations that continue to set
AWS’s services and features apart in the industry. As a member of the UC
organization, you’ll support the development and management of Compute,
Database, Storage, Platform, and Productivity Apps services in AWS, including
support for customers who require specialized security solutions for their cloud
services. Additionally, this role may involve exposure to and experience with
Amazon's growing suite of generative AI services and other cloud computing
offerings across the AWS portfolio.

About the team
AWS Neuron is the software of Trainium and Inferentia, the AWS Machine Learning
chips. Inferentia delivers best-in-class ML inference performance at the lowest
cost in the cloud to our AWS customers. Trainium is designed to deliver the
best-in-class ML training performance at the lowest training cost in the cloud,
and it’s all being enabled by AWS Neuron. Neuron is a Software that include ML
compiler and native integration into popular ML frameworks. Our products are
being used at scale with external customers like Anthropic and Databricks as
well as internal customers like Alexa, Amazon Bedrocks, Amazon Robotics, Amazon
Ads, Amazon Rekognition and many more.

Requirements

• PhD, or Master's degree and 4+ years of CS, CE, ML or related field experience
• Experience in patents or publications at top-tier peer-reviewed conferences or journals
• Experience programming in Java, C++, Python or related language
• Experience in any of the following areas: algorithms and data structures, parsing, numerical optimization, data mining, parallel and distributed computing, high-performance computing

Nice To Haves

• Experience in state-of-the-art deep learning models architecture design and deep learning training and optimization and model pruning
• Experience building machine learning models or developing algorithms for business application

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

What This Job Offers

• Job Type: Full-time
• Career Level: Mid Level
• Education Level: Ph.D. or professional degree