New Grad - ML Stack Optimization Engineer

Cerebras · Semiconductors · Headquarters +2 · Software

New Grad ML Stack Optimization Engineer role at Cerebras, focusing on optimizing compiler technologies for AI chips using LLVM and MLIR frameworks to enhance performance and efficiency of AI applications on their wafer-scale architecture.

What you'd actually do

  1. Design, develop, and optimize compiler technologies for AI chips using LLVM and MLIR frameworks.
  2. Identify and address performance bottlenecks, ensuring optimal resource utilization and execution efficiency.
  3. Work with the machine learning team to integrate compiler optimizations with AI frameworks and applications.
  4. Contribute to the advancement of compiler technologies by exploring new ideas and approaches.

Skills

Required

  • C/C++ programming
  • low-level optimization
  • Python programming
  • optimization techniques
  • NP-hard problems

Nice to have

  • Satisfiability Problem
  • Integer-Linear Programming
  • Constraint Satisfaction Problems
  • MLIR

Other signals

  • Optimizing compiler technologies for AI workloads
  • Leverage the unique capabilities of our CS3 system
  • Achieving unprecedented levels of performance, efficiency, and scalability for AI applications