Compiler Engineer
About this role:
As compiler engineer, you will be responsible for developing compiler optimizations for our state-of-the-art spatial compiler - targeting Groq's revolutionary Tensor Streaming Processor. You will own specific components and compiler passes within Groq's TSP compiler, and be in charge of designing new optimizations, 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 your components and passes produce 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 & opportunities in this role:
- Design, develop, and maintain key components and passes within Groq's TSP compiler
- Propose and expand Groq IR dialect to reflect the ever changing landscape of ML constructs and models.
- Benchmark and analyze output produced by optimizing compiler, and quantify quality-of-results when measured on the Groq TSP hardware.
- Assist in the publication of novel compilation techniques to Groq's TSP at top-tier ML, Compiler, and Computer Architecture conferences.
Ideal candidates have/are:
- Strong initiative and self starter
- Keen attention to detail
- Strong written and oral communication; ability to write clear and concise technical documentation
- Team first attitude
Qualifications for this role:
- A degree in computer science, computer engineering, or related field
- 1+ year of experience with C/C++ or Python programming
- Knowledge of functional programming an asset
- Experience with distributed systems or spatial compute such as FPGAs
- Experience with ML frameworks such as TensorFlow or PyTorch desired
- Knowledge of ML IR representations such as ONNX and Deep Learning
Attributes of a Groqster:
- Humility - Egos are checked at the door
- Collaborative & Team Savvy - We make up the smartest person in the room, together
- Growth & Giver Mindset - Learn it all versus know it all, we share knowledge generously
- Curious & Innovative - Take a creative approach to projects, problems, and design
- Passion, Grit, & Boldness - no limit thinking, fueling informed risk taking
If this sounds like you, we’d love to hear from you!
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.