about 2 months ago

Logo of  Groq

Staff Software Engineer, Speculative Decoding

$175k - $307k

Groq

Mountain View, CARemote

About Groq

Groq delivers fast, efficient AI inference. Our LPU-based system powers GroqCloud™, giving businesses and developers the speed and scale they need. Headquartered in Silicon Valley, we are on a mission to make high performance AI compute more accessible and affordable. When real-time AI is within reach, anything is possible. Build fast.

Staff Software Engineer, Speculative Decoding

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.

 

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:  talent@groq.com.  This contact information is for accommodation requests only.  Evaluation of requests for reasonable accommodations will be determined on a case-by-case basis.