Next-Gen AI Performance Engineer - Kernel & WSE Optimization

Cerebras Systems; Toronto; On-site; CAD 120,000 - 180,000; Full time

Job summary

Cerebras Systems is seeking a performance-focused engineer to characterize, analyze, and optimize AI models on the Wafer Scale Engine hardware. You will work across hardware and software to identify bottlenecks and improve efficiency, influencing Cerebras’ next-generation AI architecture and software systems.

Responsibilities include bringing up new hardware generations, building end-to-end performance models, and debugging runtime performance while developing visualization tools for performance

Qualifications

• Bachelor’s or higher degree in Electrical Engineering or Computer Science.
• Strong background in computer architecture.
• Exposure to low-level deep learning / LLM math.
• Analytical and problem-solving mindset.
• 3+ years of related experience (Computer Architecture, CPU/GPU Performance, Kernel Optimization, HPC).
• Experience working on CPU/GPU simulators.
• Comfort with C++ and Python.

Responsibilities

• Bring up and optimize performance on new generations of the Cerebras WSE.
• Build performance models (kernel-level, end-to-end) to estimate performance of state-of-the-art models.
• Optimize and debug kernel micro code and compiler algorithms to elevate inference speed, throughput and compute utilization on the Cerebras WSE.
• Debug and understand runtime performance on the system and cluster.
• Develop tools and infrastructure to visualize performance data from the Wafer Scale Engine and compute cluster.

Education

Bachelors / Masters / PhD in Electrical Engineering or Computer Science

Strong background in computer architecture

Exposure to low-level deep learning / LLM math

Analytical and problem-solving mindset

3+ years of experience in Computer Architecture, CPU/GPU Performance, Kernel Optimization, HPC

Experience with CPU/GPU simulators

C++ and Python

Tools

CPU/GPU simulators

Job description

Cerebras Systems is seeking a performance-focused engineer to characterize, analyze, and optimize AI models on the Wafer Scale Engine hardware. You will work across hardware and software to identify bottlenecks and improve efficiency, influencing Cerebras’ next-generation AI architecture and software systems.

Responsibilities include bringing up new hardware generations, building end-to-end performance models, and debugging runtime performance while developing visualization tools for performance