Senior Research Engineer - Inference ML

As a Senior Research Engineer on the Inference ML team at Cerebras Systems, you will adapt today's most advanced language and vision models to run efficiently on our flagship Cerebras architecture. You'll work alongside ML researchers and engineers to design, prototype, validate, and optimize models, gaining end-to-end exposure to cutting-edge inference research on the world's fastest AI accelerator.

You will focus on pushing the frontier of speculative decoding, large-model pruning and compression, sparse attention, and sparsity-driven techniques to deliver low-latency, high-throughput inference at scale.

• Design, implement, and optimize state-of-the-art transformer architectures for NLP and computer vision on Cerebras hardware.
• Research and prototype novel inference algorithms and model architectures that exploit the unique capabilities of Cerebras hardware, with emphasis on speculative decoding, pruning/compression, sparse attention, and sparsity.
• Train models to convergence, perform hyperparameter sweeps, and analyze results to inform next steps.
• Bring up new models on the Cerebras system, validate functional correctness, and troubleshoot any integration issues.
• Profile and optimize model code using Cerebras tools to maximize throughput and minimize latency.
• Develop diagnostic tooling or scripts to surface performance bottlenecks and guide optimization strategies for inference workloads.
• Collaborate across teams, including software, hardware, and product, to drive projects from inception through delivery.

Minimum Qualifications

…