CoDesign & NextGen Performance Engineer

Job summary

Responsibilities

This role focuses on characterizing, analyzing, and optimizing the performance of state-of-the-art AI models running on Cerebras’ breakthrough hardware. You will work across the hardware and software stack to identify bottlenecks, improve computational efficiency, and help influence the design of…

• Bring up and optimize performance on new generations of the Cerebras WSE.
• Build performance models (kernel-level, end-to-end) to estimate the performance of state of the art and customer ML models.
• Optimize and debug our kernel micro code and compiler algorithms to elevate ML model 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 help visualize performance data collected from the Wafer Scale Engine and our compute cluster.

What they're looking for

• Bachelors / Masters / PhD in Electrical Engineering or Computer Science.
• Exposure to and understanding of low-level deep learning / LLM math.
• Strong analytical and problem-solving mindset.
• 3+ years of experience in a relevant domain (Computer Architecture, CPU/GPU Performance, Kernel Optimization, HPC).
• Experience working on CPU/GPU simulators.

Tech stack

• C++
• Python