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. We are known for pioneering brands like Firefox, the privacy-minded web browser, and Pocket, a service for keeping up with the best content online. With more than 225 million people around the world using our products each month, we are committed to reclaiming an internet built for people, not companies. Our work focuses on diverse areas including AI, social media, security, and more, while maintaining our core mission to make the internet better for people.

About This Role

We are looking for a Staff Machine Learning Engineer to join our Firefox team. This team is a community of engineers dedicated to delivering the fastest, friendliest, and most usable browser possible. As a Staff Machine Learning Engineer, you will help us develop and grow new Generative AI-driven products and tools. You will play a key role in enabling safe and healthy machine learning and Generative AI-driven 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 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.
• Previous contributions to open-source projects.
• Experience with responsible AI, transparent algorithms, and putting users’ needs first.
• Experience with the browser extensions architecture or ecosystem.

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.

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 ensure that qualified individuals with disabilities are provided reasonable accommodations to participate in the job application or interview process, to perform essential job functions, and to receive other benefits and privileges of employment, as appropriate.

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