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. We are seeking exceptional software engineers to join our compiler team in tackling the exciting problem of translating ML programs written in existing ML frameworks into the language of our custom Groq ML accelerator chips.
We are looking for deep learning specialists to be based in either our Menlo Park, CA or Portland, OR office.
ESSENTIAL DUTIES & RESPONSIBILITIES:
Optimize model performance: Leverage your knowledge of machine learning workloads to identify compiler optimizations that maximize program execution efficiency on Groq hardware platforms.
Design compiler features: Work with members of the Groq’s compiler team to design and evaluate potential features for our compiler. Collaborate closely with hardware designers to identify ways to take advantage of hardware capabilities.
Manage performance tradeoffs: Understand the balance between performance, memory, and power in compiler generated code.
Innovate: Contribute to the design of next generation hardware architectures and compiler releases.
Collaborate with others through code reviews, design discussions, and problem-solving sessions.
SKILLS & QUALIFICATIONS:
Master’s degree in computer science or mathematics, or equivalent practical experience
3+ years experience developing and optimizing deep learning models
Strong coding skills
Experience with TensorFlow and other common frameworks (ONNX, Caffe, PyTorch)
Excellent collaboration skills, with strong verbal and written communication
PhD in computer science or mathematics
Familiarity with functional programming languages (such as Haskell)
Understanding of compiler architecture and compiler implementation techniques
Ability to build strong relationships
Good written and oral communication skills; strong technical documentation skills
Good interpersonal skills; a history of mentoring junior engineers and interns a huge plus
Highly self-motivated and directed; self-confidence and self-starter
Keen attention to detail
Proven analytical and problem-solving abilities
Ability to effectively prioritize and execute tasks in a high-pressure environment
Experience working in a team-oriented, collaborative environment