← All roles

ML Runtime and Kernel Engineer - Core ML

Cerebras· Software Engineering
Apply on company site
Hybrid (Sunnyvale, CA), Hybrid (Toronto, CAN)Full-timePosted 3w ago

About the role

Role focused on bridging research into high-performance Core ML software on Cerebras Wafer-Scale Engine, spanning runtime components, high-performance kernels, ML frameworks, compilers, and low-level optimizations for large-scale training and low-latency inference.

Responsibilities

  • Design and implement runtime components and high-performance kernels required by novel Core ML algorithms.
  • Translate research prototypes into efficient implementations for the Cerebras platform, including reference implementations and comparisons on GPUs where useful.
  • Profile and debug performance across the ML framework, compiler, runtime, communication, and kernel layers.
  • Optimize computation, memory movement, communication, and concurrency for large-scale training and low-latency inference.
  • Develop benchmarks, instrumentation, and automated tests that validate functionality, performance, and numerical correctness.
  • Collaborate closely with Core ML researchers and compiler, runtime, kernel, and inference engineers to evaluate design alternatives and deliver end-to-end capabilities.
  • Contribute to software architecture and roadmap decisions by identifying recurring limitations and high-leverage platform improvements.

Qualifications

  • 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.
  • Proven ability to debug and profile complex software across multiple layers of a system.
  • Familiarity with modern machine learning architectures and frameworks such as PyTorch or JAX.
  • Ability to work effectively with researchers and translate evolving algorithmic requirements into reliable software.

Skills

Required:High-performance systems software developmentC++PythonParallel programmingDebugging and profilingML frameworks (PyTorch or JAX)
Last updated Oct 4, 2026. Interested? Apply on the company site.

About Cerebras

Cerebras is the go-to platform for fast and effortless AI training. Learn more at cerebras.ai.

Cerebras is the go-to platform for fast and effortless AI training.

AICerebras Wafer-Scale Enginechipshardwareinferenceliquid-ai-recruiting-targetmultimodalsetter-30
Company profileWebsite

More open roles at Cerebras