Senior Applied Scientist, Neuron ARG

The AWS Neuron Compiler team's Automated Reasoning Group is seeking a Senior Applied Scientist to work at the intersection of AI and program analysis. This role focuses on enhancing code quality in their advanced deep learning compiler stack, which optimizes applications across various domains including Large Language and Vision models from frameworks like PyTorch, TensorFlow, and JAX. The position involves working with AWS's custom ML accelerators - Inferentia and Trainium, which are crucial for their Generative AI capabilities. The role combines cutting-edge research with practical implementation, requiring expertise in compiler technology, program analysis, and deep learning systems. The successful candidate will lead development of compiler analyzers, publish research, and mentor team members while working with state-of-the-art ML infrastructure. This position offers competitive compensation ranging from $150,400 to $260,000 based on location, plus comprehensive benefits and the opportunity to work on pioneering technology at AWS.