Sr. Compiler Engineer

Details

Groq is a machine learning systems company building easy-to-use solutions for accelerating artificial intelligence workloads. Our work spans hardware, software, and machine learning technology. As Sr. Compiler Engineer, you will be responsible for defining and developing compiler optimizations for our state-of-the-art spatial compiler - targeting Groq's revolutionary Tensor Streaming Processor. You will be the technical lead for Groq's TSP compiler, and be in charge of architecting new passes, developing innovative scheduling techniques, and developing new front-end language dialects to support the rapidly evolving ML space. You will also be required to benchmark and monitor key performance metrics to ensure that the compiler is producing efficient mappings of neural network graphs to the Groq TSP. Experience with LLVM and MLIR preferred, and knowledge with functional programming languages an asset. Also, knowledge with ML frameworks such as TensorFlow and PyTorch, and portable graph models such as ONNX desired.

Responsibilities

• Design, develop, and maintain optimizing compiler for Groq's TSP
• Expand Groq IR Dialect to reflect ever changing landscape of ML constructs and models
• Benchmark and analyze output produced by optimizing compiler, and drive enhancements to improve its quality-of-results when measured on the Groq TSP hardware.
• Manage large multi-person and multi-geo projects and interface with various leads across the company
• Mentor junior compiler engineers and collaborate with other senior compiler engineers on the team.
• Review and accept code updates to compiler passes and IR definitions.
• Work with HW teams and architects to drive improvements in architecture and SW compiler
• Publish novel compilation techniques to Groq's TSP at top-tier ML, Compiler, and Computer Architecture conferences.

Requirements

• 10+ years of experience in the area of computer science/engineering or related
• 5+ years of direct experience with C/C++ and LLVM or compiler frameworks
• Knowledge of spatial architectures such as FPGA or CGRAs an asset
• Knowledge of functional programming an asset
• Experience with ML frameworks such as TensorFlow or PyTorch desired
• Knowledge of ML IR representations such as ONNX and Deep Learning

Benefits

Compensation: At Groq, a competitive base salary is part of our comprehensive compensation package, which includes equity and benefits. For this role, the base salary range is $143,600 to $251,300, determined by your skills, qualifications, experience and internal benchmarks.