Staff Machine Learning Engineer
You will lead the development and optimization of Large Language Models and Mixture of Experts models. You will deliver product initiatives involving on-device inference optimization, quantization, retrieval-augmented generation, agentic frameworks, and tool calling. You will work cross-functionally to integrate models into the platform, conduct research to improve model performance and efficiency, and apply advances in AI and machine learning. You will lead complex projects, mentor junior engineers, share knowledge and best practices, and lead a sub-team.
Responsibilities
- Lead the development and optimization of Large Language Models and Mixture of Experts models
- Collaborate with cross-functional teams to integrate machine learning models into the platform
- Conduct machine learning research to improve the performance and efficiency of Large Language Models
- Apply advances in AI and machine learning to improve models and methodologies
- Mentor junior engineers and contribute to knowledge sharing and best practices
- Lead core product initiatives across on-device inference optimization, quantization, retrieval-augmented generation, agentic frameworks, and tool calling
- Deliver projects end-to-end with other engineering functions
- Lead the sub-team
Requirements
- Advanced degree in Computer Science or a related field; Ph.D. preferred
- Proven record of building and innovation through publications or industry experience
- At least 6 years of machine learning experience with expertise in Large Language Models and Mixture of Experts
- Strong Python programming skills
- Experience with TensorFlow and/or PyTorch
- Ability to lead complex projects and collaborate in a team environment
Benefits
- Comprehensive health, dental, and vision benefits package
- 401(k) match
- Equity options
- $200/month Health & Wellness stipend
- Continuing Education support
- $500/year Function Health subscription
- Free parking for in-office employees
- Flexible Time Off
- Parental leave for eligible employees
- Supplemental life insurance