Position Details

Type Full-Time

Experience senior

Exp. Years Not specified

Education Not specified

Category AI & Machine Learning

About this role

Amazon’s Annapurna ML team is hiring a Senior Applied Scientist to improve code quality and compiler tooling for deep learning systems. The role focuses on building and deploying analyzers for ML compiler stages and compiler IRs, working closely with accelerators and research-minded engineering teams.

Key Responsibilities

Design and implement analyzers for ML compiler stages and compiler IRs
Architect and implement business-critical tooling
Publish research and mentor scientists/engineers
Collaborate on leading-edge compiler and deep-learning systems software
Innovate on behalf of customers to improve the compiler stack

Technical Overview

You will design, develop, and deploy analyzers for ML compiler stages and compiler IRs within a deep learning compiler stack. The work connects program analysis and compiler technology with generative AI workflows and accelerator platforms including Inferentia and Trainium, using frameworks such as PyTorch, TensorFlow, and JAX.

Ideal Candidate

The ideal candidate is a senior-level applied scientist who has hands-on experience with compilers and program analysis for machine learning systems, including working with compiler IRs and ML compiler stages. They are comfortable designing and deploying analyzers and tooling for deep learning compiler stacks and can communicate research outcomes to technical and engineering stakeholders. Experience with machine learning accelerators such as Inferentia and Trainium is a strong differentiator.

Must-Have Skills

program analysisdeep learning compiler stackdesigning analyzers for ML compiler stagesarchitecting and implementing tooling for compiler IRsprogramming languagescompilersprogram analyzersprogram synthesis engines

Nice-to-Have Skills

background in machine learning and AI accelerators

Tools & Platforms

PyTorchTensorFlowJAX

Required Skills

Artificial Intelligenceprogram analysiscode qualitydeep learning compiler stackMachine Learning acceleratorsInferentiaTrainiumGenerative AIanalyzersML compiler stagescompiler IRscompilersprogram analyzersprogram synthesis enginesPyTorchTensorFlowJAX

Hard Skills

Artificial Intelligenceprogram analysiscode qualitydeep learning compiler stackmachine learning acceleratorsInferentiaTrainiumgenerative AIanalyzersML compiler stagescompiler IRscompiler technologydeep-learning systems softwareprogramming languagescompilersprogram analyzersprogram synthesis enginesdesigningdevelopingdeployingpublishing researchtooling

Soft Skills

mentoringteam leadershipcommunicationcuriositycollaborationexecuting in a research-focused environmentworking closely with a team of scientists and engineersinnovationcustomer focus

Industry & Role

Industry Aerospace

Job Function Build and deploy compiler analyzers and tooling for ML workloads on Amazon’s Annapurna accelerator stack

Role Subtype AI Researcher

Tech Domains Python, JavaScript, AI & Machine Learning, Amazon Web Services

Keywords for Your Resume

Senior Applied ScientistApplied ScientistArtificial Intelligenceprogram analysiscode qualitydeep learning compilercompiler IRsML compiler stagescompiler technologyprogramming languagescompilersprogram analyzersprogram synthesis enginesMachine Learning acceleratorsInferentiaTrainiumGenerative AIPyTorchTensorFlowJAXresearchtoolingpublishing research

Deal Breakers

Must demonstrate experience with compilers and program analysis (program analyzers / program synthesis engines)., Must have experience designing and deploying tooling/analyzers for ML compiler stages or compiler IRs.

Apply for this Position →

Get matched to jobs like this

Luna finds roles that fit your skills and career goals — no endless scrolling required.

Create a Free Profile

Senior Applied Scientist, Annapurna ML

Get matched to jobs like this