Staff Machine Learning Engineer, Gen AI Remote Belgium

About Mozilla

Mozilla Corporation is the non-profit-backed technology company that has shaped the internet for the better over the last 25 years. We make pioneering brands like Firefox, the privacy-minded web browser, and Pocket, a service for keeping up with the best content online.

Our work focuses on diverse areas including AI, social media, security and more. And we’re doing this while never losing our focus on our core mission – to make the internet better for people. The Role

We are looking for a Staff Machine Learning Engineer to 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, collaborating cross functionally with product management, full stack engineering and design.
• Build infrastructure for training and inference of LLMs and small language and vision models for use cases on the 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 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.
• Leading architecture designs to execution independently and track record of improving the team’s overall AI development velocity.
• 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. Additional Information

We are an equal opportunity employer and welcome applications from everyone, including members of all equity-seeking communities. We believe that valuing diverse creative practices and forms of knowledge are crucial to and enrich the company’s core mission.

Commitment to diversity, equity, inclusion, and belonging: Mozilla understands that valuing diverse creative practices and forms of knowledge are crucial to and enrich the company’s core mission. We encourage applications from everyone, including members of all equity-seeking communities, such as (but certainly not limited to) women, racialized and Indigenous persons, persons with disabilities, persons of all sexual orientations, gender identities, and expressions.