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, as we are wholly owned by the Mozilla Foundation. Our flagship product, the Firefox browser, is part of a family of products that protect users and provide a secure internet 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.

What You'll Bring

• 6+ years of experience as a Machine Learning engineer.
• Experience with designing and building machine learning tooling and infrastructure.
• Experience with distributed systems and platforms for AI integrations.
• Strong problem-solving skills and ability to communicate complex concepts to technical and non-technical stakeholders.
• Experience working collaboratively with product managers and 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 everyone, including members of all equity-seeking communities. 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.