Applied Scientist - ML Systems & Compiler (Neuron)

Amazon; San Francisco (CA); On-site; USD 136,000 - 223,400; Full time

Job summary

A leading technology company in California seeks talented scientists for the AWS Neuron Science Team. You will enhance their software stack and collaborate with customers to identify barriers and optimize solutions for Trainium and Inferentia accelerators. Qualified candidates should have a PhD in computer science or related fields, experience in algorithms, and proficiency in programming languages like Java, C++, and Python. This role offers a competitive salary and the chance to work with innovative ML technologies.

Qualifications

• PhD in computer science, computer engineering, or 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, high-performance computing.
• Experience using Unix/Linux.

Responsibilities

• Develop and apply ML/RL approaches for kernel/code generation.
• Create advanced compiler techniques for ML workloads.
• Build tools for accuracy and reliability validation.
• Design high-performance kernels optimized for ML accelerator architectures.

Skills

Java programming

C++ programming

Python programming

Algorithms and data structures

Numerical optimization

Parallel computing

Data mining

Experience with MXNet

Experience with TensorFlow

Unix/Linux usage

Education

PhD in computer science or computer engineering

Job description

A leading technology company in California seeks talented scientists for the AWS Neuron Science Team. You will enhance their software stack and collaborate with customers to identify barriers and optimize solutions for Trainium and Inferentia accelerators. Qualified candidates should have a PhD in computer science or related fields, experience in algorithms, and proficiency in programming languages like Java, C++, and Python. This role offers a competitive salary and the chance to work with innovative ML technologies.