Drive AI acceleration as an ML Kernel Performance Engineer with AWS Neuron at Annapurna Labs. Leverage your expertise in kernel optimization and low-level performance tuning to maximize deep learning workloads.

AWS Neuron at Annapurna Labs is seeking an experienced ML Kernel Performance Engineer to work at the convergence of hardware and software. In this role, you’ll design high-performance compute kernels, analyze kernel-level performance, and collaborate with global teams to optimize machine learning workloads. Your contributions are vital in providing customers with optimal performance on AWS ML accelerators.

Key Responsibilities:
• Design high-performance compute kernels for ML operations
• Optimize kernel-level performance across various Neuron hardware




• Conduct performance analysis using profiling tools
• Implement compiler optimizations for improved efficiency
• Collaborate directly with customers on model optimization

Requirements:
• 3+ years of skilled software development experience
• 2+ years in system design or architecture experience
• Familiarity with one programming language
• Bachelor's degree in computer science or equivalent preferred
• Experience with full software development life cycle

Shape the future of AI acceleration by optimizing ML models at AWS Neuron.
#J-18808-Ljbffr

📌 ML Kernel Engineer at AWS Neuron (Ontario)
🏢 Amazon
📍 Ontario