Cerebras Systems
Staff Kernel Optimzation Engineer
Remote, California, United States · Staff+
Sponsorship not specifiedDetected 61 days ago
PythonC++AlgorithmsAPI DevelopmentMachine LearningDeep LearningTensorFlowLLMsCompliance
About the role
- Your focus will be on implementing, optimizing, and scaling deep learning operations to fully leverage our custom, massively parallel processor architecture.
- Your work will be critical to unlocking the full potential of our hardware and accelerating the pace of AI innovation.
Responsibilities
- 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.
- Interact with chip and system architects to optimize instruction sets, microarchitecture, and IO of next generation systems.
Nice to have
- Preferred Skills And Qualifications
Skills
- 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.
- 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.
- Five Reasons to Join Cerebras in 2026.
- Cerebras Systems is committed to creating an equal and diverse environment and is proud to be an equal opportunity employer.
Benefits
- Develop design specifications for new machine learning and linear algebra kernels and mapping to the Cerebras WSE System using various parallel programming algorithms.
- 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.
Company info
- Build a breakthrough AI platform beyond the constraints of the GPU.
- Publish and open source their cutting-edge AI research.
- Work on one of the fastest AI supercomputers in the world.
- Enjoy job stability with startup vitality.
- Our simple, non-corporate work culture that respects individual beliefs.
Apply directly at Cerebras Systems →Create a free account for alerts like thisView Cerebras Systems immigration profile
This listing is sourced directly from Cerebras Systems's careers page and normalized into a canonical job model.