ML Runtime and Kernel Engineer - Core ML

Join Cerebras as an ML Runtime and Kernel Engineer to develop novel machine learning algorithms on our groundbreaking Wafer-Scale Engine. You will bridge the gap between research and efficient execution, working across ML frameworks, compilers, and low-level kernels to implement new capabilities and optimize performance for training and inference.

• Design and implement ML runtime components and high-performance kernels
• Translate research prototypes into efficient implementations for Cerebras systems
• Profile and debug performance across ML framework, compiler, runtime, and kernel layers
• Optimize computation, memory, communication, and concurrency for large-scale AI
• Collaborate with researchers and engineers to deliver end-to-end capabilities
• Develop benchmarks, instrumentation, and automated tests