Applied Scientist II, Annapurna ML

About the role

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.

Responsibilities

• Work directly with external and internal customers to identify key adoption barriers and optimization opportunities.
• Collaborate closely with our engineering teams to implement innovative solutions.
• Engage with academic and research communities to advance state-of-the-art ML systems.
• 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.
• Exposure to and experience with Amazon's growing suite of generative AI services and other cloud computing offerings across the AWS portfolio.

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.

Qualifications

• 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 applications.

Skills

• Strong problem-solving skills.
• Excellent communication and collaboration skills.
• Ability to work independently and as part of a team.

Benefits

AWS offers a comprehensive benefits program including health insurance, retirement savings plans, paid time off, and parental leave. For more information, visit our benefits page.

Pay

$171,600.00 - $222,200.00 USD annually

Schedule

Full-time