Cerebras
ML Runtime and Kernel Engineer - Core ML
Sunnyvale, CA +1 · Hybrid · Full-time · Posted Sep 28, 2026
- Location
- Sunnyvale, CA
Toronto, CAN - Education
- Bachelor’s, Master’s, PhD, or equivalent practical experience in Computer Science, Computer Engineering, Electrical Engineering, or a related field.
About the job
Develop runtime components and high-performance kernels that bring novel machine-learning algorithms to efficient execution on Cerebras systems.
What you'll do
- Implement runtime components and kernels, and translate research prototypes into efficient platform implementations.
- Profile and optimize performance across framework, compiler, runtime, communication, and kernel layers.
- Build benchmarks, instrumentation, and automated tests; collaborate with researchers and engineering teams on end-to-end capabilities.
What they ask for
- Bachelor’s, Master’s, PhD, or equivalent practical experience in Computer Science, Computer Engineering, Electrical Engineering, or a related field.
- Experience developing high-performance systems software, ML systems, runtimes, compilers, or computational kernels.
- Strong programming skills in C++ and Python.
- Solid understanding of parallel programming, memory management, concurrency, data structures, and performance optimization.
Nice to have
- Experience with CUDA, Triton, low-level assembly, accelerator programming, or a C-like domain-specific language.
- Experience with compiler internals, distributed runtimes, custom hardware interfaces, or HPC systems.
Tools
C++, Python, PyTorch, JAX, CUDA, Triton
AI focus
Implement and optimize runtime and kernel capabilities for Core ML algorithms, including LLM training and inference, parallel and diffusion-based generation, sparsity, and distributed execution.
Model-written summary · the original posting is authoritative