01 Nov
|
Amazon
|
Toronto
Optimize machine learning kernels for Amazon Neuron using low-level performance analysis and accelerator expertise in Toronto, Ontario.As a Senior Machine Learning Kernel Engineer, you will develop high-performance kernels for Inferentia and Trainium accelerators. You will work across compiler, runtime, framework, and hardware teams. The work includes model enablement, profiling, compiler optimization, and customer support for machine learning workloads.Key Responsibilities: Design high-performance kernels for machine learning operationsAnalyze performance across Neuron hardware generationsUse profiling tools to resolve kernel bottlenecksImplement fusion, sharding, tiling, and scheduling optimizationsEnable and optimize customer models on AWS acceleratorsRequirements: Five-plus years of skilled software development experienceFive-plus years programming with a software languageFive-plus years leading system design or architectureExperience mentoring or leading engineering teamsExperience with accelerator architectures or GPU optimizationBring kernel development, compiler optimization, and machine learning acceleration expertise to Amazon Neuron in Toronto.#J-18808-Ljbffr
📌 Senior Machine Learning Kernel Engineer (Toronto)
🏢 Amazon
📍 Toronto