Amazon Applied Scientist AWS Neuron Science Team 

United States, California 

9877648

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 architecturesAbout the team
Diverse Experiences
AWS values diverse experiences. Even if you do not meet all of the preferred qualifications and skills listed in the job description, we encourage candidates to apply. If your career is just starting, hasn’t followed a traditional path, or includes alternative experiences, don’t let it stop you from applying.Mentorship & Career Growth
We’re continuously raising our performance bar as we strive to become Earth’s Best Employer. That’s why you’ll find endless knowledge-sharing, mentorship and other career-advancing resources here to help you develop into a better-rounded professional.Work/Life Balance

- PhD, or a Master's degree and experience building machine learning models or developing algorithms for business application
- Experience in patents or publications at top-tier peer-reviewed conferences or journals
- Experience in any of the following areas: algorithms and data structures, parsing, numerical optimization, data mining, parallel and distributed computing, high-performance computing
- Experience programming in Java, C++, Python or related language

- Experience using Unix/Linux
- Experience in professional software development