Skip to content
Cerebras SystemsCerebras SystemsSunnyvale, CA

Advanced Technology: R&D Engineer - AI/ML, HPC

Designs and implements AI/ML and scientific computing workloads on wafer-scale hardware to set performance benchmarks. Leads algorithm-hardware co-design, builds performance models, and contributes to technology roadmap with publications in top venues. PhD preferred in CS, Engineering, or related field.

Salary not listed
On-siteML Engineering

About the role

What You Will Do

  • Design and implement challenging scientific computing and AI workloads on Cerebras' Wafer-Scale Engine, targeting performance results that advance the state of the art.
  • Lead algorithm–hardware co-design efforts with internal R&D teams and external research partners, turning architectural capabilities into measurable application-level advantages.
  • Build analytical performance models that quantify bottlenecks, guide optimization, and inform future chip and compiler design decisions.
  • Contribute to Cerebras' multi-year technology roadmap by identifying high-impact workloads, proposing architectural experiments, and validating them on silicon.
  • Publish findings and present at top-tier conferences and journals; represent Cerebras in the broader HPC and AI research communities.

What We Are Looking For

  • PhD in Computer Science, Engineering, Applied Mathematics, Physics, or a related quantitative field preferred. Exceptional candidates without a graduate degree who demonstrate equivalent depth through published research, significant open-source contributions, or a strong industry track record are encouraged to apply.
  • Deep experience in at least one of the following: computer architecture and accelerator design; parallel, distributed, or high-performance computing; numerical methods and scientific simulation; AI/ML theory and model design at a mathematical level.
  • Strong ability to analytically model and optimize the performance of complex systems and algorithms.
  • Track record of published research or patents in relevant venues.
  • Proficiency in C and Python; comfort working close to hardware.
  • Excellent communication and interpersonal skills: able to present complex technical material to both specialist and cross-functional audiences, and to collaborate effectively in a fast-paced, small-team environment.

Areas Of Particular Interest

We are hiring across several focus areas. Exceptional depth in one or more of the following is a strong signal:

  • Computational science: researchers who can bring insights from numerical methods and simulation into AI, or couple simulation and learning into joint computational workflows. Depth in hydrodynamics, solid mechanics, electromagnetics, molecular dynamics, or related PDE-based fields.
  • AI/ML foundations: deep understanding of model architecture, optimization methods, and their statistical underpinnings—the ability to design from first principles, not just apply established recipes.
  • Computer architecture: microarchitecture design, computing paradigms at the circuit and datapath level, memory hierarchy design.
  • Performance engineering: roofline modeling, bandwidth analysis, kernel optimization, communication-computation overlap, and compiler-level tuning for novel hardware.

Skills

AI/MLHpcCPythonComputer ArchitectureNumerical MethodsScientific SimulationPerformance ModelingParallel ComputingCompiler Optimization

Similar roles

ML Engineering jobs
Cloudflare

Software Engineer, AI Agents

CloudflareUnited States

Build and ship production AI agents on Cloudflare's edge platform using Workers, Durable Objects, and AI tools. Requires strong TypeScript/Rust experience, observability expertise, and hands-on LLM tooling for evals, safety, and multi-agent systems.

Salary not listed
On-siteML Engineering
Cerebras Systems

CoDesign & NextGen Performance Engineer

Cerebras SystemsSunnyvale, CA

Characterize, analyze, and optimize performance of state-of-the-art AI models on Cerebras' wafer-scale hardware. Build performance models, optimize kernels and compilers, debug runtime behavior, and develop visualization tools to influence next-gen AI architecture.

Salary not listed
On-site3+ YOEML Engineering
OpenAI

Research Engineer, Privacy

OpenAISan Francisco, CA

Research Engineer on OpenAI's Privacy team designing and prototyping privacy-preserving ML algorithms like differential privacy and federated learning at scale. Requires hands-on PETs experience, fluency in PyTorch/JAX, and a track record implementing or publishing novel privacy work.

380k – 445k/yr
HybridML Engineering
Console

Research Engineer

ConsoleSan Francisco, CA

Research Engineer building self-improving AI agent systems at Console. Develop eval/optimization loops, fine-tune specialist models, and improve agent reasoning over enterprise context using production data to drive measurable gains in quality, latency, and reliability.

200k – 350k/yr
On-siteML Engineering
Notion

Software Engineer, AI Platform

NotionSan Francisco, CA +1

Build and scale the shared AI platform foundations at Notion, enabling fast and safe shipping of AI products. Requires experience with LLM/ML platforms, strong ownership, and comfort across backend, infrastructure, and product code.

180k – 201k/yr
Hybrid5+ YOEML Engineering