Core ML Performance & Mapping Engineer

Cerebras; Sunnyvale (CA); On-site; USD 180,000 - 260,000; Full time

Job summary

Cerebras Systems is seeking an engineer for the Core ML team to map novel algorithms to our architecture and benchmark their performance at scale. You will combine modeling, benchmarking, and prototyping to map efficiency frontiers for ML training and inference on Cerebras hardware.

You will influence which research ideas proceed, implement prototypes on Cerebras systems, and help define capabilities for future hardware and software generations.

Qualifications

• Experience with analytical performance modeling and benchmarking.

Responsibilities

• Build analytical and empirical performance models for state-of-the-art ML training and inference algorithms.
• Characterize asymptotic behavior and identify how algorithmic trade-offs change with model size, sequence length, batch size, parallelism, and hardware scale.
• Construct Pareto frontiers across model quality, latency, throughput, memory footprint, communication, and compute cost.
• Develop prototype implementations and benchmarks for Cerebras WSE and relevant GPU or software baselines.
• Analyze system behavior to identify kernel, compiler, runtime, communication, and algorithmic bottlenecks.
• Evaluate emerging techniques in areas such as parallel token generation, diffusion and speculative decoding, attention, sparsity, mixture-of-experts, low-precision computation, and distributed training.
• Partner with researchers and kernel, compiler, runtime, inference, and architecture teams to recommend high-value implementation and co-design directions.
• Develop tools and visualizations that make performance projections, measurements, and design trade-offs understandable across engineering and research teams.
• Clearly communicate conclusions, assumptions, limitations, and recommendations through technical reports, presentations, and design reviews.

Skills

Python

C++

Performance modeling

Benchmarking

Parallel computing

Machine learning systems

Analytical reasoning

Communication

Education

Bachelor's, Master's, or PhD in CS/CE/Math or related field

Tools

CUDA

Triton

PyTorch

JAX

Job description

Cerebras Systems is seeking an engineer for the Core ML team to map novel algorithms to our architecture and benchmark their performance at scale. You will combine modeling, benchmarking, and prototyping to map efficiency frontiers for ML training and inference on Cerebras hardware.

You will influence which research ideas proceed, implement prototypes on Cerebras systems, and help define capabilities for future hardware and software generations.