Applied Scientist — ML Systems & Compiler (Neuron)

Amazon Web Services (AWS)

Santa Clara (CA)

On-site

USD 172,000 - 222,000

Full time

3 days ago

Be an early applicant

Get more replies from employers

Send a job-specific resume in minutes.

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.