Applied Scientist (AI)

TL;DR

Applied Scientist, AWS Neuron Science Team (AI): Enhancing AWS Neuron's software stack to accelerate customer adoption of Trainium and Inferentia accelerators with an accent on identifying key adoption barriers and optimization opportunities. Focus on implementing innovative solutions, creating advanced compiler techniques, and advancing state-of-the-art ML systems through collaboration with engineering, academic, and research communities.

Location: Onsite in Santa Clara, CA, USA. Specific legal information for Los Angeles County applicants is also provided.

Salary: $171,600.00 – $222,200.00 USD annually

Company

Amazon Web Services (AWS) is the world's most comprehensive and broadly adopted cloud platform, pioneering cloud computing and innovating for startups and Global 500 companies.

What you will do

• Enhance AWS Neuron's software stack to accelerate customer adoption of Trainium and Inferentia accelerators.
• Work directly with external and internal customers to identify key adoption barriers and optimization opportunities.
• Collaborate closely with engineering teams to implement innovative solutions.
• Engage with academic and research communities to advance state-of-the-art ML systems.
• Develop and apply ML/RL approaches for kernel/code generation and optimization.
• Create advanced compiler techniques for ML workloads.

Requirements

• PhD in computer science, computer engineering, or a related field.
• Experience in patents or publications at top-tier peer-reviewed conferences or journals.
• Experience in algorithms and data structures, parsing, numerical optimization, data mining, parallel and distributed computing, or high-performance computing.
• Programming experience in Java, C++, or Python.
• Experience using Unix/Linux.

Nice to have

• Experience in investigating, designing, prototyping, and delivering new and innovative system solutions.
• Experience in state-of-the-art deep learning models architecture design, deep learning training and optimization, and model pruning.
• Experience with popular deep learning frameworks such as MxNet and TensorFlow.

Culture & Benefits

• Work in a strategic growth area for AWS alongside distinguished engineers and scientists.
• Inclusive team culture fostered by employee-led and company-sponsored affinity groups.
• Opportunities for continuous knowledge-sharing, mentorship, and career advancement.
• Commitment to work-life harmony and flexibility.
• Comprehensive benefits including health insurance (medical, dental, vision, prescription), 401(k) matching, paid time off, and parental leave.