Applied Scientist — ML Systems & Compiler (Neuron)

Amazon Web Services (AWS); Santa Clara (CA); On-site; USD 172,000 - 222,000; Full time

Benefits offered by this job

Health insurance

RSUs / equity participation

401(k) matching

Job summary

Annapurna Labs (U.S.) Inc. in Santa Clara, CA is seeking highly qualified scientists to advance ML systems for the Trainium and Inferentia stack.

You will develop ML/RL approaches for kernel and code generation and help optimize performance across ML workloads. You will collaborate with internal teams and external customers, contributing to state-of-the-art ML compiler technologies and robust system validation, while engaging with research communities to push the boundaries of cloud-based AI

Qualifications

• PhD in computer science, computer engineering, or related field.
• Experience in patents or publications at top-tier conferences or journals.
• Experience in algorithms, 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.

Responsibilities

• Develop ML/RL approaches for kernel/code generation and optimization.
• Create advanced compiler techniques for ML workloads.
• Collaborate with engineering teams to implement innovative ML system solutions.
• Engage with academic and research communities to advance state-of-the-art ML systems.

Skills

Java

C++

Python

Unix/Linux

Education

PhD in Computer Science/Engineering

Tools

MxNet

TensorFlow

Job description

Annapurna Labs (U.S.) Inc. in Santa Clara, CA is seeking highly qualified scientists to advance ML systems for the Trainium and Inferentia stack.

You will develop ML/RL approaches for kernel and code generation and help optimize performance across ML workloads. You will collaborate with internal teams and external customers, contributing to state-of-the-art ML compiler technologies and robust system validation, while engaging with research communities to push the boundaries of cloud-based AI