Groq Logo

Groq

Principal Inference Stack Engineer

Posted Yesterday
Be an Early Applicant
Remote
Hiring Remotely in Toronto, ON
Expert/Leader
Remote
Hiring Remotely in Toronto, ON
Expert/Leader
Lead efforts in optimizing ML workloads for Groq’s LPU through deep knowledge of the compiler and ML inference stack. Responsibilities include designing and maintaining optimizations, managing projects, mentoring junior staff, and collaborating with hardware teams.
The summary above was generated by AI

At Groq. We believe in an AI economy powered by human agency. We envision a world where AI is accessible to all, a world that demands processing power that is better, faster, and more affordable than is available today. AI applications are currently constrained by the limitations of the Graphics Processing Unit (GPU), a technology originally developed for the gaming market and soon to become the weakest link in the AI economy.

Enter Groq's LPU™ AI Inference Technology. Specifically engineered for the demands of large language models (LLMs), the Language Processing Unit outpaces the GPU in speed, power, efficiency, and cost-effectiveness. The quickest way to understand the opportunity is to watch the following talk – groq.link/scspdemo.

Why join Groq? AI will change humanity forever, and we believe preservation of human agency and self determination is only possible if AI is made affordably and universally accessible. Groq’s LPUs will power AI from an early stage, and you will get to leave your fingerprint on civilization.

Principal Inference Stack Engineer

Mission: lead technical efforts focused on mapping ML workloads onto Groq’s LPU through deep and first hand working knowledge of Groq’s state-of-the-art spatial compiler and ML inference stack.

Responsibilities & outcomes:

  • Analyze latest ML workloads from Groq partners or Cloud and develop optimization roadmap and strategies to improve inference performance and operating efficiency of workload
  • Design, develop, and maintain optimizing compiler for Groq's LPU
  • Expand Groq runtime API to simplify execution model of Groq LPUs
  • Benchmark and analyze output produced by optimizing compiler and runtime, and drive enhancements to improve its quality-of-results when measured on the Groq LPU 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, Applications, Compiler, and Computer Architecture conferences.

Ideal candidates have/are:

  • 10+ years of experience in the area of computer science/engineering or related
  • 5+ years of direct experience with C/C++ and runtime frameworks
  • Knowledge of LLVM and compiler architecture
  • Experience with mapping HPC, ML, or Deep Learning workloads to accelerators
  • Knowledge of spatial architectures such as FPGA or CGRAs an asset
  • Knowledge with distributed systems and disaggregated compute desired
  • Knowledge of functional programming an assert
  • 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 $248,710 to $407,100, determined by your skills, qualifications, experience and internal benchmarks.

Location: Groq is a geo-agnostic company, meaning you work where you are. Exceptional candidates will thrive in asynchronous partnerships and remote collaboration methods. Some roles may require being located near our primary sites, as indicated in the job description.  

At Groq: Our goal is to hire and promote an exceptional workforce as diverse as the global populations we serve. Groq is an equal opportunity employer committed to diversity, inclusion, and belonging in all aspects of our organization. We value and celebrate diversity in thought, beliefs, talent, expression, and backgrounds. We know that our individual differences make us better.


Groq is an Equal Opportunity Employer that is committed to inclusion and diversity. Qualified applicants will receive consideration for employment without regard to race, color, religion, national origin, gender, sexual orientation, gender identity, disability or protected veteran status.  We also take affirmative action to offer employment opportunities to minorities, women, individuals with disabilities, and protected veterans.

Groq is committed to working with qualified individuals with physical or mental disabilities. Applicants who would like to contact us regarding the accessibility of our website or who need special assistance or a reasonable accommodation for any part of the application or hiring process may contact us at:  [email protected].  This contact information is for accommodation requests only.  Evaluation of requests for reasonable accommodations will be determined on a case-by-case basis.

Top Skills

C
C++

Similar Jobs

5 Hours Ago
3 Locations
Remote
80 Employees
Entry level
80 Employees
Entry level
Artificial Intelligence • Machine Learning • Software
As a Software Engineer at mabl, you will design and maintain cloud systems for mobile app testing, collaborate on product solutions, and support the integration of customer feedback while ensuring quality across the application lifecycle.
Be an Early Applicant
13 Hours Ago
Canada
Remote
2,800 Employees
Senior level
2,800 Employees
Senior level
Artificial Intelligence • Cloud • Computer Vision • Hardware • Internet of Things • Software
As a Senior Embedded Software Engineer on Samsara's Firmware Team, you'll design and develop embedded platforms for camera systems, work on custom hardware and software development, debug complex systems, mentor engineers, and collaborate with various teams to improve product quality and efficiency.
Be an Early Applicant
13 Hours Ago
Canada
Remote
2,800 Employees
Mid level
2,800 Employees
Mid level
Artificial Intelligence • Cloud • Computer Vision • Hardware • Internet of Things • Software
The Senior Quality Engineer will advocate for quality in the firmware quality pod, define quality goals, create and maintain test suites, prioritize automation of tests, run acceptance tests, diagnose issues, mentor junior engineers, and promote a culture of quality analytics.

What you need to know about the Toronto Tech Scene

Although home to some of the biggest names in tech, including Google, Microsoft and Amazon, Toronto has established itself as one of the largest startup ecosystems in the world. And with over 2,000 startups — more than 30 percent of the country's total startups — Toronto continues to attract new businesses. Be it helping entrepreneurs manage their finances, simplifying business operations by automating payroll or assisting pharmaceutical companies in launching new drugs, the city's tech scene is just getting started.

Sign up now Access later

Create Free Account

Please log in or sign up to report this job.

Create Free Account