Senior ML Kernel Engineer — Neuron Performance

Amazon; Toronto; On-site; CAD 90,000 - 120,000; Full time

Job summary

A global technology company in Toronto is seeking a Kernel Engineer to design high-performance kernels for machine learning operations. The role involves collaboration across software, hardware, and system layers to optimize performance for AWS's ML accelerators. Candidates should have experience in system architecture and ML model acceleration, with a focus on delivering cutting-edge solutions to customers.

Benefits

Flexible working hours

Mentorship and career growth

Inclusive team culture

Qualifications

• 3+ years of non-internship professional software development experience.
• Experience programming with at least one software programming language.
• Knowledge of ML frameworks (PyTorch, TensorFlow) and their GPU backends.

Responsibilities

• Design and implement high-performance compute kernels for ML operations.
• Analyze and optimize kernel-level performance across multiple generations.
• Conduct detailed performance analysis using profiling tools.

Skills

Low-level optimization

System architecture

ML model acceleration

Performance analysis

Education

Bachelor's degree in computer science or equivalent

Tools

CUDA

OpenCL

LLVM

Job description

A global technology company in Toronto is seeking a Kernel Engineer to design high-performance kernels for machine learning operations. The role involves collaboration across software, hardware, and system layers to optimize performance for AWS's ML accelerators. Candidates should have experience in system architecture and ML model acceleration, with a focus on delivering cutting-edge solutions to customers.