Staff Machine Learning Engineer, Gen AI at Mozilla

Mozilla Corporation is seeking a Staff Machine Learning Engineer to join the Firefox team. This role focuses on developing and growing new Generative AI-driven products and tools within the Firefox browser. The engineer will play a crucial role in enabling safe and healthy machine learning and Generative AI experiences.

The role involves leading the design, development, and integration of Generative AI solutions in Firefox, collaborating with product management, full-stack engineering, and design teams. The engineer will build infrastructure for training and inference of LLMs and small language and vision models for use cases on web and mobile Firefox product experiences. Also, they will implement robust validation and testing procedures to ensure the developed models' generalizability and reliability, and continuously monitor and optimize deployed models for performance and efficiency.

What this role involves:

• Leading the design, development, and integration of Generative AI solutions in Firefox.
• Building infrastructure for training and inference of LLMs.
• Implementing robust validation and testing procedures.
• Monitoring and optimizing deployed models for performance and efficiency.

Requirements:

• 6+ years of experience as a Machine Learning engineer building tooling and services for machine learning applications in production.
• Experience with designing and building machine learning tooling and infrastructure for training, deploying, inference and validation of models.
• Experience with distributed systems and platforms for AI integrations.
• Strong problem-solving skills and ability to communicate complex concepts.
• Experience working collaboratively with product managers and non-engineering teams.
• Effective documentation and communication skills on a distributed team.
• Commitment to Mozilla's values.

What Mozilla offers:

• Generous performance-based bonus plans.
• Rich medical, dental, and vision coverage.
• Generous retirement contributions.
• Quarterly all-company wellness days.
• Country-specific holidays plus a day off for your birthday.
• One-time home office stipend.
• Annual professional development budget.
• Quarterly well-being stipend.
• Considerable paid parental leave.
• Employee referral bonus program.