Advanced Technology: Ai/ml Research Scientist

Cerebras · Semiconductors · Headquarters +3 · Advanced Technology

Research Scientist role focused on designing AI models and training methods from first principles, leveraging novel wafer-scale hardware architectures. The role involves investigating computational science techniques for AI, understanding hardware-algorithm interactions, and publishing research at top-tier venues. The work directly influences future hardware and software design.

What you'd actually do

  1. Design AI models and training methods from first principles, leveraging architectural properties of wafer-scale hardware that are unavailable on conventional platforms.
  2. Investigate how techniques from computational science—numerical methods, PDE solvers, simulation—can inform and advance AI model design, and explore hybrid workflows that couple simulation and learning.
  3. Develop a deep understanding of the hardware substrate and use it to guide algorithmic choices: model structure, optimization strategy, memory access patterns, numerical precision.
  4. Publish findings and present at top-tier venues (NeurIPS, ICML, ICLR, etc.); represent Cerebras in the broader AI/ML research community.
  5. Inform the design of future Cerebras hardware and software by identifying the computational patterns that matter most for next-generation AI workloads.

Skills

Required

  • Python
  • PyTorch
  • Mathematical maturity
  • Theory behind gradient methods, loss landscapes, generalization, and the relationship between model structure and data statistics

Nice to have

  • C or other low-level languages

What the JD emphasized

  • PhD in Machine Learning, Computer Science, Applied Mathematics, Statistics, Physics, or a related quantitative field preferred
  • Track record of published research at top-tier AI or computational science venues.

Other signals

  • design AI models and training methods from first principles
  • leveraging architectural properties of wafer-scale hardware
  • publish findings and present at top-tier venues
  • inform the design of future Cerebras hardware and software