Staff Machine Learning Software Engineer
About this role:
We seek exceptional machine learning developers/ engineers/researchers with experience developing machine learning models. As a part of the ML Systems team at Groq, you will work closely with Groq's sales, applications, and engineering teams to develop and optimize ML models and systems for our hardware and contribute to original research in the field.
Responsibilities & opportunities in this role:
- Develop kernels and models (both customer and public domain models) for Groq hardware using low-level proprietary frameworks and popular higher-level machine learning frameworks. The models will span domains from machine learning (computer vision, natural language processing, recommendation engines, reinforcement learning) to high-performance computing (linear algebra)
- Optimize models for Groq’s hardware by exploiting proprietary hardware features
- Performance analysis of models on Groq hardware
- Performance analysis of large-scale systems built using Groq hardware
- Performance analysis of models on competitor hardware/systems
- Contribute to driving features into Groq’s hardware based on model optimizations/insight
- Publish research papers on ML model optimizations and hardware in top-tier ML conferences.
Ideal candidates have/are:
- Analytical background with the ability to quickly understand complex hardware technologies, understand tradeoffs and build systems using them.
- ML (Neural Networks) and math fundamentals expertise, with some experience/deep understanding in one or more of the following areas:
- Computer vision
- Natural Language Processing
- Recommendation engines
- Reinforcement Learning
- Linear algebra
- Experience with a subset of the following:
- Python and C/C++
- TensorFlow, Pytorch, Caffe, or other ML Frameworks
- HW accelerator programming languages such as CUDA, MKLDNN
- Programming experience on other accelerators such as FPGAs or DSPs from evaluation to production.
- Understanding of processor architectures and distributed systems and their implications on ML model performance
- A strong writer and public speaker, I operate with integrity and drive transparency, openness, and effective communication.
Qualifications for this role:
- Min MS in CS, CE/EE, Math, and Physics.
- 5 to 10 years of software and machine learning experience.
- Recent Ph.D. computer science, math, or engineering graduates with ML experience will also be considered.
- Publication record in ML conferences (ICML, NeurIPS, ICLR, CVPR) is a plus
- Excellent leadership, mentoring, and cross-functional collaborative and influencing skills.
- Effective communication & presentation skills
- Able to work in a very dynamic start-up environment
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 $289,100, determined by your skills, qualifications, experience and internal benchmarks.