Staff Machine Learning Engineer

Lead the development and optimization of Large Language Models and Mixture of Experts models, deliver initiatives involving inference optimization, quantization, RAG, agentic frameworks, and tool calling, integrate models into the platform, conduct research, mentor engineers, 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 model performance and efficiency
• Apply advances in AI and machine learning to improve models and methodologies
• Mentor junior engineers and contribute to knowledge sharing and best practices
• Lead 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
• Excellent problem-solving skills and passion for innovation

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