Staff Kernel Optimzation Engineer
Cerebras · Remote
About this role
## About Cerebras Cerebras Systems builds the world’s largest AI chip—an architecture designed to deliver industry-leading training and inference speeds (over 10× faster than GPU-based hyperscale cloud inference services). The work helps enable real-time iteration and additional agentic computation for smarter AI applications. Cerebras partners with leading model labs, global enterprises, and AI-native startups, including a recent multi-year partnership with OpenAI to deploy 750 megawatts of scale. --- ## About the Role As a **Kernel Engineer**, you’ll develop high-performance software solutions at the intersection of hardware and software. You will implement, optimize, and scale deep learning operations to fully leverage Cerebras’ massively parallel processor architecture. You’ll be part of a world-class team responsible for the design, performance tuning, and validation of foundational ML and HPC kernels—building a library of parallel and distributed algorithms that maximize compute utilization and improve training efficiency. --- ## Responsibilities - Develop design specifications for new **machine learning** and **linear algebra** kernels and map them to the **Cerebras WSE System** using parallel programming algorithms. - Develop and debug a kernel library of highly optimized low-level assembly and **C-like domain specific language (DSL)** routines. - Develop and debug high-performance kernel routines in low-level assembly and **CSL** (custom C-like language), implementing algorithms optimized for Cerebras hardware. - Use mathematical models and analysis to measure software performance and guide design decisions. - Develop and integrate unit and system testing methodologies to verify correctness and performance. - Study emerging ML trends and help evolve kernel library architecture to address computational challenges in state-of-the-art neural networks. - Collaborate with chip and system architects to optimize instruction sets, microarchitecture, and IO for next-generation systems. --- ## Skills & Qualifications - Bachelor’s, Master’s, PhD, or foreign equivalents in **Computer Science, Computer Engineering, Mathematics**, or related fields. - Understanding of **hardware architecture concepts** and comfort learning a new hardware architecture. - Strong programming skills in **C++** and **Python**. - Knowledge of **library/API development best practices**. - Strong debugging skills, including experience debugging complex software stacks. --- ## Preferred Qualifications - Experience with kernel development and/or testing. - Familiarity with parallel algorithms and distributed memory systems. - Experience programming accelerators such as **GPUs** and **FPGAs**. - Familiarity with ML neural networks and frameworks such as **TensorFlow** and **PyTorch**. - Familiarity with **HPC kernels** and performance optimization. --- ## Why Join Cerebras People who are serious about software make their own hardware. Cerebras is building a breakthrough architecture that’s unlocking new opportunities for the AI industry. Team members cite five reasons to join: 1. Build a breakthrough AI platform beyond GPU constraints. 2. Publish and open-source cutting-edge AI research. 3. Work on one of the fastest AI supercomputers in the world. 4. Enjoy job stability with startup vitality. 5. A simple, non-corporate culture that respects individual beliefs. --- ## Equal Opportunity Cerebras Systems is committed to creating an equal and diverse enviro
Listing freshness
CronJobs last confirmed this listing 2h ago. If its source stops confirming the opening for seven days, this page is removed from active inventory.