Staff Kernel Optimization Engineer
Cerebras Systems, Inc.Visit Cerebras Systems, Inc. website
Cerebras builds wafer-scale AI computing systems and a cloud inference platform for training, fine-tuning, and serving AI models.
Sunnyvale, California, United States
About Cerebras Systems, Inc.
Cerebras Systems is an AI-infrastructure company founded in 2015. It sells rack-scale wafer-scale computing systems and provides cloud-based, API-accessible AI inference alongside on-premises deployments.
Skills
About the Role
You will develop, debug, optimize, and test low-level machine learning and linear algebra kernels for custom hardware. You will use performance analysis to guide design decisions, evolve kernel-library architecture for neural networks, and collaborate with chip and system architects on future systems.
Requirements
- Computer Science, Computer Engineering, Mathematics, or related degree
- Hardware architecture knowledge
- C++
- Python
- Library and API development
- Debugging complex software stacks
Responsibilities
- Develop machine learning and linear algebra kernel specifications and mappings
- Develop and debug optimized assembly and custom language kernel routines
- Measure software performance using mathematical models and analysis
- Develop unit and system testing for kernel libraries
- Study machine learning trends and evolve kernel-library architecture
- Collaborate with chip and system architects on instruction sets, microarchitecture, and I/O