Kernel Engineer for AI Hardware & HPC

Cerebras; Sunnyvale (CA); Presencial; USD 180.000 - 240.000; Jornada completa

Descripción de la vacante

Cerebras Systems builds the world''s largest AI chip and is advancing high-performance software for cutting-edge AI and HPC workloads. As a Kernel Engineer, you will implement, optimize, and scale deep learning operations to fully leverage our massively parallel processor architecture.

You will join a world-class team responsible for design, performance tuning, and validation of foundational ML and HPC kernels, creating a library of parallel and distributed algorithms to maximize compute

Formación

• Bachelor’s, Master’s, PhD or foreign equivalents in Computer Science, Computer Engineering, Mathematics, or related fields.
• Understanding of hardware architecture concepts — must be comfortable learning the details of a new hardware architecture.
• Skilled in C++ and Python programming languages.
• Good knowledge of library and/or API development best practices.
• Strong debugging skills and knowledge of debugging complex software stack.

Responsabilidades

• Develop design specifications for new machine learning and linear algebra kernels and mapping to the Cerebras WSE System using various parallel programming algorithms.
• Develop and debug kernel library of highly optimized low level assembly instruction and C-like domain specific language routines to implement algorithms targeting the Cerebras hardware system.
• Develop and debug high-performance kernel routines in low-level assembly and a custom C-like (CSL) language, implementing algorithms optimized for the Cerebras hardware system.
• Using mathematical models and analysis to measure the software performance and inform design decisions.
• Develop and integrate unit and system testing methodologies to verify correct functionality and performance of kernel libraries.
• Study emerging trends in Machine Learning applications and help evolve Kernel library architecture to address computational challenges of the start-of-the-art Neural Networks.
• Interact with chip and system architects to optimize instruction sets, microarchitecture, and IO of next generation systems.

Conocimientos

C++

Python

Hardware architecture concepts

Debugging

Library/API development

Educación

Bachelor’s/Master’s/PhD in CS/CE/Math

Herramientas

CSL language

Low-level assembly

Descripción del empleo

Cerebras Systems builds the world''s largest AI chip and is advancing high-performance software for cutting-edge AI and HPC workloads. As a Kernel Engineer, you will implement, optimize, and scale deep learning operations to fully leverage our massively parallel processor architecture.

You will join a world-class team responsible for design, performance tuning, and validation of foundational ML and HPC kernels, creating a library of parallel and distributed algorithms to maximize compute