Staff Kernel Optimzation Engineer
Cerebras · Remote
CHECKED 7H AGO
Infrastructureremotestaff
At a glance
Highlights
- Develop high-performance kernels for the world's largest AI chip.
- Maximize compute utilization to accelerate AI innovation.
- Work on one of the fastest AI supercomputers in the world.
Ideal candidate
A performance-obsessed systems thinker who loves working at the hardware-software boundary and pushing massively parallel architectures to their limits to accelerate cutting-edge AI.
Requirements
- Bachelor's, Master's, PhD or equivalent in Computer Science, Computer Engineering, Mathematics or related field
- Understanding of hardware architecture concepts and ability to learn new hardware architecture
- Skilled in C++ and Python programming
- Good knowledge of library and API development best practices
- Strong debugging skills for complex software stacks
- Experience in kernel development and testing
- Familiarity with parallel algorithms and distributed memory systems
- Experience programming accelerators such as GPUs and FPGAs
- Familiarity with ML neural networks and frameworks like TensorFlow and PyTorch
- Familiarity with HPC kernels and optimization
Lab profile
About Cerebras
Builds wafer-scale AI chips and cloud systems for ultra-fast training and inference.
What they build
- Wafer-scale AI chips
- AI training systems
- Cloud inference service
Known for
- 56x larger chips than GPUs
- 10x faster inference than GPU clouds
- Industry-leading training speed