Staff Machine Learning Engineer, Generative AI

About Mozilla

Mozilla Corporation is a non-profit-backed technology company that has been shaping the internet for the better over the last 25 years. Our mission is to ensure the internet remains open and accessible. We are not beholden to shareholders, allowing us to focus on making the internet better for people. Our flagship product, the Firefox browser, is part of a family of products that protect users and provide a secure online experience.

About the Role

We are looking for a Staff Machine Learning Engineer to join our Firefox team. This role involves developing and growing new Generative AI-driven products and tools. You will play a key role in enabling safe and healthy machine learning and Generative AI experiences in Firefox.

Responsibilities

• Lead the design, development, and integration of Generative AI solutions in Firefox.
• Collaborate cross-functionally with product management, full-stack engineering, and design teams.
• Build infrastructure for training and inference of LLMs and small language and vision models for web and mobile Firefox product experiences.
• Implement robust validation and testing procedures to ensure the developed models' generalizability and reliability.
• Continuously monitor and optimize 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 across a breadth of applications such as LLM APIs, Cloud-based, and open-source tooling.
• Strong problem-solving skills and ability to communicate complex concepts to technical and non-technical stakeholders.
• Experience working collaboratively with product managers, project managers, and/or other non-engineering teams.
• Effective documentation and communication skills on a distributed team.

Bonus Points

• Knowledge of and experience with using LLM solutions across different cloud providers.
• Experience with responsible AI, transparent algorithms, and putting users’ needs first.
• Experience with the browser extensions architecture or ecosystem.
• Experience working in open-source environments.

What You’ll Get

• Generous performance-based bonus plans.
• Rich medical, dental, and vision coverage.
• Generous retirement contributions with 100% immediate vesting.
• 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.
• Other benefits (life/AD&D, disability, EAP, etc.).

Commitment to Diversity

Mozilla is committed to diversity, equity, inclusion, and belonging. We encourage applications from all equity-seeking communities, including women, racialized and Indigenous persons, persons with disabilities, and persons of all sexual orientations, gender identities, and expressions. We provide reasonable accommodations to qualified individuals with disabilities.

Join us to work on the Web as the platform and help create more opportunity and innovation for everyone online.