Applied Scientist (AI)

TL;DR

Applied Scientist (AI): Developing and optimizing ML systems and software stack for AWS Trainium and Inferentia accelerators with an accent on AI for systems, ML compiler techniques, system robustness, and efficient kernel development. Focus on designing state-of-the-art deep learning models, compiler innovations, and collaborating with internal and external customers to overcome adoption barriers and optimize performance.

Location: Onsite in the United States

Salary: $136,000–$223,400 per year

Company

hirify.global is part of Amazon Web Services, a leading global cloud platform innovating in AI and machine learning hardware and software.

What you will do

• Enhance AWS Neuron software stack for Trainium and Inferentia ML accelerators.
• Collaborate with internal and external customers to identify adoption barriers and optimization opportunities.
• Develop AI for systems, ML compiler techniques, and efficient kernel designs.
• Work with engineering teams to implement innovative ML solutions.
• Engage with academic and research communities to advance ML systems.

Requirements

• PhD in computer science, computer engineering, or related field
• Experience with algorithms, data structures, numerical optimization, parallel and distributed computing.
• Proficiency in Java, C++, Python, and Unix/Linux environments.
• Publications or patents in top-tier peer-reviewed venues.
• Must be located in the United States for onsite work

Nice to have

• Experience designing and delivering innovative system solutions.
• Expertise in deep learning model architecture, training, optimization, and pruning.
• Familiarity with deep learning frameworks such as MxNet and TensorFlow.

Culture & Benefits

• Inclusive and diverse team culture promoting curiosity and connection.
• Mentorship and career growth opportunities.
• Work-life balance with flexible working culture.
• Comprehensive compensation including equity and benefits.