πŸ‡ΊπŸ‡Έ USAJobs.work

America's Job Portal

← Back to USA Jobs

AWS Neuron Kernel Optimization Engineer

Company

Amazon

Location

toronto, on

Posted

May 29, 2026

Position Overview

Enhance machine learning performance as a Kernel Performance Engineer at Annapurna Labs, AWS. Focus on developing high-performance compute kernels for leading-edge ML accelerators like Inferentia and Trainium.

This position seeks candidates with over 3 years of software development exposure, ideally targeting optimized ML functions. You'll analyze kernel performances, work cross-functionally on innovative solutions, and facilitate customer ML model performance on AWS platforms.

Key Responsibilities:
β€’ Design and implement kernels for machine learning applications
β€’ Optimize kernel performance across different hardware
β€’ Conduct profiling and analysis to address issues
β€’ Apply compiler optimizations for improved efficiency
β€’ Collaborate with various teams to drive technical innovations

Requirements:
β€’ 3+ years in professional software development
β€’ 2+ years in system design or architecture
β€’ Knowledge...

Ready to Apply?

Join thousands of Americans building their careers

Apply Now