Applied Scientist, SSG Science

About the Role

Amazon Devices is an inventive research and development company that designs and engineers high-profile devices. This is an exciting opportunity to join Amazon in developing state-of-the-art techniques that bring Gen AI on edge for our consumer products. We are looking for exceptional scientists to join our Applied Science team and help develop the next generation of edge models, and optimize them while doing co-design with custom ML HW based on a revolutionary architecture.

Key Responsibilities

• Quantize, prune, distill, finetune Gen AI models to optimize for edge platforms
• Fundamentally understand Amazon’s underlying Neural Edge Engine to invent optimization techniques
• Analyze deep learning workloads and provide guidance to map them to Amazon’s Neural Edge Engine
• Use first principles of Information Theory, Scientific Computing, Deep Learning Theory, Non Equilibrium Thermodynamics
• Train custom Gen AI models that beat SOTA and pave the path for developing production models
• Collaborate closely with compiler engineers, fellow Applied Scientists, Hardware Architects, and product teams to build the best ML-centric solutions for our devices
• Publish in open source and present on Amazon's behalf at key ML conferences - NeurIPS, ICLR, MLSys

Requirements & Tech Stack

• Experience programming in Java, C++, Python or related language

Qualifications

• Bachelor's degree in computer science, mathematics, statistics, machine learning or equivalent quantitative field
• Experience implementing algorithms using both toolkits and self-developed code
• Have publications at top-tier peer-reviewed conferences or journals