Sr. Machine Learning - Compiler Engineer III, AWS Neuron, Annapurna Labs

What you'll do

• Design, implement, test, deploy, and maintain compiler software and performance, stability, and usability improvements.
• Develop compiler optimization and verification passes, analysis tools, and features that expose accelerator capabilities to developers.
• Collaborate with chip, runtime, OS, science, and ML teams and engage with open-source communities and partners.

What they ask for

• Bachelor's degree in computer science or equivalent.
• 5+ years of full software development life cycle experience, including coding standards, code reviews, source control management, build processes, testing, and operations.
• 3+ years of experience developing compiler features and optimizations.
• Proficiency in one or more of C++ (preferred) or Python.

Nice to have

• Master or PhD degree in computer science or equivalent.
• Experience with compiler design and related optimization areas, LLVM and/or MLIR, or LLM, vision, or other deep-learning models.

Pay

193,300.00 - 261,500.00 USD annually (base salary); 168,100.00 - 227,400.00 USD annually (base salary)

Tools

AWS Neuron, AWS Inferentia, Trainium, C++, Python, LLVM, MLIR, OpenXLA

AI focus

Optimize compilation and deployment of machine-learning models, including large language models, diffusion, and vision models, for AWS accelerators.