Staff Software Engineer, Speculative Decoding Mountain View, CA (Remote)

Groq Inc.; Mountain View (CA); Remote; USD 175,000 - 308,000; Full time

Job summary

Join a forward-thinking company at the forefront of AI technology, where you will play a crucial role in developing innovative algorithms that enhance generative AI capabilities. As a Staff Software Engineer specializing in Speculative Decoding, you will leverage your expertise in C++ and Rust to design and implement cutting-edge solutions that improve the speed and accuracy of AI inference. This role offers the opportunity to work in a collaborative environment, driving impactful projects from conception to production. If you are passionate about AI and eager to make a significant contribution to the future of technology, this position is perfect for you.

Benefits

Equity options

Comprehensive benefits package

Flexible working hours

Remote work opportunities

Qualifications

• Master's degree or equivalent experience required.
• Hands-on experience in generative AI inference is essential.

Responsibilities

• Design and optimize speculative decoding algorithms for AI.
• Collaborate with teams to integrate solutions into production.

Skills

Generative AI Inference

Problem-solving

Analytical Skills

Collaboration

Creativity

Education

Master’s degree in Computer Science

Electrical Engineering

Tools

C++

Rust

Kubernetes

MPI

PyTorch

Job description

At Groq, we believe AI will change humanity forever, and that making it affordable and universally accessible is the key to human agency in an AI economy. We’re assembling a team of world-class engineers and business minds who believe in this mission. We’re hiring problem-solvers with the skills and desire to build a business that leaves a fingerprint on civilization.

Staff Software Engineer, Speculative Decoding

Mountain View, CA (Remote)

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.

Mission: with a strong background in Generative AI Inference and expertise in Speculative Decoding, you will design, implement, and optimize cutting-edge algorithms to enhance our production AI infrastructure and capabilities in post training, model evaluation, and operational performance.

Responsibilities & outcomes:

• Design, implement, and optimize speculative decoding algorithms and underlying models that enhance the speed and accuracy of Generative AI Inference.
• Collaborate with cross-functional teams to integrate your solutions into Groq’s production AI infrastructure.
• Work in a multi data center production environment and Kubernetes environment with Groq’s customer hardware, inference and compiler stack.
• Develop high-performance, scalable code primarily in C++ and Rust, ensuring efficient resource utilization and system stability. Ability to model performance of a distributed high performance system.
• Experience building production distributed systems involving multi process communication with technologies such as MPI, scheduling and working in a Kubernetes environment.
• Stay up-to-date with the latest developments in generative AI and speculative decoding, and translate cutting-edge research into practical, production-ready implementations.
• Work closely with teams across software engineering, research, and operations to drive improvements in post training, model evaluation, and overall system performance.
• Provide technical leadership and mentorship to team members, fostering an environment of continuous learning and innovation.
• Champion code quality, maintainability, observability, monitoring and best practices, ensuring that all deliverables meet rigorous performance and security standards.

Ideal candidates have/are:

• Master’s degree in Computer Science, Electrical Engineering, or a related field (or equivalent industry experience).
• Extensive, hands-on experience in generative AI inference with a specific focus on speculative decoding.
• Proficiency in C++ is essential, with demonstrated experience in developing high-performance systems.
• Strong analytical and problem-solving skills, with a track record of delivering innovative technical solutions.
• Proven ability to work effectively in fast-paced, cross-functional environments, driving projects from conception to production.
• Understanding of the architecture of Generative AI models, PyTorch, familiarity with the data science necessary to evaluate layers of models, their performance and quality.
• Familiarity with AI infrastructure challenges and scalable system design.

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 $175,900 to $307,800, 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.