Description
AWS is hiring a Kernel Engineer for its Neuron Compiler organization to design and implement high-performance compute kernels for machine learning operations on Inferentia and Trainium accelerators. The role involves optimizing kernel-level performance, using profiling tools, applying compiler optimizations such as fusion, sharding, tiling, and scheduling, and collaborating with customers and cross-functional teams. Candidates need at least three years of professional software development experience, two years of design or architecture experience, and experience with at least one programming language; a bachelor's degree in computer science or equivalent is preferred. The position is based in Cupertino, California, with a base salary of $165,200 to $223,600 annually.
