# Staff Kernel Optimization Engineer

**Company:** [Cerebras Systems](https://hotfix.jobs/companies/cerebras-systems)
**Location:** Remote
**Role:** Embedded Engineering
**Experience:** 7+ years
**Skills:** C++, Python, Kernel Development, Parallel Algorithms, Distributed Memory Systems, Gpu Programming, Fpga Programming, TensorFlow, PyTorch, Hpc Kernels, Low-Level Assembly, Debugging
**Posted:** 2026-02-05

> Develop and optimize high-performance ML and HPC kernels in low-level assembly and CSL for Cerebras' custom AI hardware architecture. Requires strong C++/Python skills and hardware architecture knowledge.

## Job Description

## Responsibilities
- 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.
- Use 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.

## Requirements
- 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.

## Preferred Qualifications
- Experience in kernel development and/or testing.
- Familiarity with parallel algorithms and distributed memory systems.
- Experience in programming accelerators such as GPUs and FPGAs.
- Familiarity with Machine Learning neural networks and frameworks such as TensorFlow and PyTorch.
- Familiarity with HPC kernels and their optimization.

## Similar roles

- [Senior Staff Software Engineer, Founding BMC](https://hotfix.jobs/jobs/3f50adc3-f9d4-4f33-a747-636ebd5b1af4) - Crusoe - San Francisco, CA - $237k – $288k/yr
- [Staff Software Engineer](https://hotfix.jobs/jobs/ab47df60-d5d4-41b8-82a9-273980251fd7) - Huntress - Remote - $200k – $220k/yr
- [Sr. Member of Technical Staff](https://hotfix.jobs/jobs/a0933979-c0fd-4514-a8e6-a1faaf186193) - Illumio - Sunnyvale, CA - $170k – $196k/yr
- [Senior / Staff Connectivity Integration Lead](https://hotfix.jobs/jobs/927394e1-aac0-4ada-b39c-e104c373eea8) - Zoox - Foster City, CA - $220k – $275k/yr
- [Staff Embedded Software Engineer](https://hotfix.jobs/jobs/bac0a112-d82c-4ca0-b162-8e41837367a4) - Cyngn - Mountain View, CA - $205k – $220k/yr

**Apply:** https://hotfix.jobs/jobs/c3954491-2364-4a24-9b4a-1209c4123f47
**Canonical:** https://hotfix.jobs/jobs/c3954491-2364-4a24-9b4a-1209c4123f47