Senior ML Kernel Optimization Engineer

About

Lead innovations in machine learning as a Senior ML Kernel Optimization Engineer. Focus on high-performance solutions for advanced ML workloads, leveraging cutting-edge technologies in a supportive environment.

As a crucial part of the kernel engineering team, you will design and implement robust kernels while optimizing existing ones. Engage with customers, spearhead performance analysis, and mentor fellow engineers in a vibrant culture of learning and inclusion. Your work will set new standards in AI acceleration.

Responsibilities

• Architect high-performance kernels for ML operations
• Optimize performance across various Neuron hardware
• Analyze performance bottlenecks with profiling tools
• Oversee compiler optimizations for better efficiency
• Collaborate with customers for model performance

Requirements

• 5+ years of software development experience
• Proficiency in at least one programming language
• Leadership in design or architecture processes
• Mentor or tech lead experience essential
• Bachelor's degree in related field preferred

Utilize your skills to redefine performance in machine learning and contribute to groundbreaking AI solutions.