Amazon
Apply Now
Lead ML Kernel Performance Engineer at Amazon
Description
Join Amazon’s Annapurna Labs team as a Lead ML Kernel Performance Engineer. Collaborate on cutting-edge technology to design high-performance computing kernels for deep learning workloads on AWS accelerators.
This role requires expertise in accelerating AI applications on custom hardware. You will work closely with multiple teams to innovate kernel optimizations for Amazon's Inferentia and Trainium accelerators. With over five years of experience in software development and architecture, you will design and implement high-impact solutions that directly influence our customer base.
Key Responsibilities:
• Design high-performance compute kernels for ML operations
• Analyze and optimize performance across Neuron hardware
• Conduct performance analysis with profiling tools
• Implement compiler optimizations such as sharding and fusion
• Collaborate with customers to optimize ML models
Requirements:
• 5+ years in sof...
This role requires expertise in accelerating AI applications on custom hardware. You will work closely with multiple teams to innovate kernel optimizations for Amazon's Inferentia and Trainium accelerators. With over five years of experience in software development and architecture, you will design and implement high-impact solutions that directly influence our customer base.
Key Responsibilities:
• Design high-performance compute kernels for ML operations
• Analyze and optimize performance across Neuron hardware
• Conduct performance analysis with profiling tools
• Implement compiler optimizations such as sharding and fusion
• Collaborate with customers to optimize ML models
Requirements:
• 5+ years in sof...