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 Mozilla Corporation is wholly owned by the not-for-profit 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 is crucial in 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.
• Commitment to Mozilla's values: welcoming differences, being relationship-minded, practicing responsible participation, and having grit.

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.

Benefits

Mozilla offers a range of benefits including generous performance-based bonus plans, rich medical, dental, and vision coverage, generous retirement contributions, and more. We also provide quarterly all-company wellness days, a one-time home office stipend, and an annual professional development budget.

Commitment to Diversity

Mozilla is committed to diversity, equity, inclusion, and belonging. We encourage applications from all equity-seeking communities and ensure that qualified individuals with disabilities are provided reasonable accommodations.

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