Description
AWS is hiring a Senior Software Engineer, Compiler for its Neuron team to build and optimize a compiler that transforms ML models from frameworks such as PyTorch, TensorFlow, and JAX into optimized code for AWS Inferentia and Trainium accelerators. The role involves compiler optimization and verification passes, accelerator feature development, numerical-error analysis, defect resolution, design discussions, code reviews, and collaboration with chip architects, runtime and OS engineers, scientists, and ML applications teams. Candidates need at least five years of leading system design or architecture and two years of compiler feature and optimization experience, with C++, C, or Python proficiency; C++ is preferred. The position is based in Cupertino, California, with a base salary of $193,300 to $261,500 annually.
