Description
AWS Neuron Compiler is hiring an ML Compiler Engineer to design, develop, test, deploy, and maintain compiler features for a deep-learning compiler stack. The role works with Inferentia/Trainium accelerators and collaborates with Runtime, Frameworks, and Hardware teams on instruction scheduling, memory allocation, data transfer optimization, graph partitioning, parallel programming, code generation, instruction set architectures, hardware bring-up, and hardware-software co-design. Candidates need a bachelor's degree in a relevant field, at least three years of professional software development experience, programming proficiency, and experience with optimization algorithms, graph theory, hardware bring-up, FPGA algorithms, or hardware resource management.
