Advanced Technology: Compiler Engineer

Cerebras Cerebras · Semiconductors · Headquarters +2 · Advanced Technology

Cerebras is seeking a Compiler Engineer to work on their Tungsten language compiler, which is purpose-built for their wafer-scale AI hardware. The role involves designing and implementing compiler passes, co-designing language constructs, and developing code generation strategies for AI and scientific workloads. The engineer will collaborate with ASIC, kernel, and AI teams, and contribute to the broader toolchain including runtime and debuggers. Experience with novel architectures and ML compiler frameworks is valuable.

What you'd actually do

  1. Design and implement compiler passes across the Tungsten toolchain: mid-end optimization, backend code generation, instruction scheduling, register allocation, assembler, and linker.
  2. Co-design language constructs that improve expressiveness and performance for dataflow execution on wafer-scale hardware.
  3. Develop and iterate on code generation strategies for complex scientific and AI workloads, analyzing performance bottlenecks and closing the gap between peak and achieved throughput.
  4. Extend the compiler to support future hardware architectures as they move from design to silicon.
  5. Work directly with ASIC architects and application researchers to inform hardware-software co-design decisions.

Skills

Required

  • Compiler development (IR design, optimization passes, code generation, backend implementation)
  • Computer architecture (instruction sets, memory models, dataflow execution)
  • Systems-level programming in C
  • Performance analysis
  • Hardware-software co-design

Nice to have

  • Compilers for spatial, dataflow, or CGRA architectures
  • ML compiler frameworks (MLIR, XLA, TVM)
  • Multi-dimensional data representations, tiling strategies, vectorized operations
  • Published research or patents in compilers, programming languages, or architecture
  • Runtime systems, debuggers, or architecture simulators
  • Parallel/distributed systems and cluster computing

What the JD emphasized

  • PhD in Computer Science or Computer Engineering preferred
  • Substantial experience in compiler development
  • Strong grasp of computer architecture
  • Systems-level programming ability in C
  • Ability to think about compilation as a design problem
  • Track record of published research or patents in compilers, programming languages, or architecture.

Other signals

  • AI chip
  • wafer-scale architecture
  • training and inference speeds
  • large-scale ML applications
  • Generative AI inference
  • AI workloads