Description
AWS is hiring a Software Engineer for its Neuron Compiler 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, accelerator feature integration, defect resolution, open-source tooling, and collaboration with chip architects, runtime and OS engineers, scientists, and ML applications teams. Candidates need at least three years of professional software development experience, two years of design or architecture experience, and experience with an object-oriented language; compiler, ML, and accelerator experience are preferred. The position is based in Cupertino, California, with a base salary of $165,200 to $223,600 annually.
