Software Development Engineer, AWS Neuron, Annapurna Labs

What you'll do

• Design, implement, test, deploy, and maintain compiler software, including optimization and verification passes.
• Improve compiler performance, stability, and usability; analyze numerical errors and resolve compiler defects.
• Collaborate with chip architects, runtime and OS engineers, scientists, ML application teams, customers, and open-source communities.

What they ask for

• 3+ years of non-internship professional software development experience.
• 2+ years of non-internship experience in design or architecture of new and existing systems, including design patterns, reliability, and scaling.
• Experience programming with at least one software programming language.

Nice to have

• Master's degree or PhD in Computer Science or a related technical field.
• Experience with compiler design, object-oriented production code, open-source compiler toolsets, ML technologies, or modern build systems.

Pay

165,200.00 - 223,600.00 USD annually (base salary)

Tools

C++, Java, OpenXLA, StableHLO, MLIR, PyTorch, JAX, Bazel/CMake

AI focus

Develop compiler optimizations for deep-learning models, including large language models, stable diffusion, and vision transformers, targeting AWS Inferentia and Trainium.