Senior Staff Machine Learning Engineer, GenAI

Mozilla
Summary
Join the Firefox team as a Senior Staff Machine Learning Engineer and help develop and grow new machine learning-driven products and tools. You will play a key role in enabling safe and healthy machine learning and AI-driven experiences in Firefox. As an industry expert, you will help define product strategy and goals, mentor junior engineers, and collaborate with management. You will tackle large open-ended problems, identify strategic opportunities for LLMs, lead the design and development of AI and machine learning models, and stay up-to-date with the latest advancements in AI. You will also lead by example to upskill the team on AI and generate thought leadership. This role requires a strong background in machine learning, experience with LLMs and multimodal models, and a deep understanding of fundamental machine learning concepts. The position offers a generous compensation and benefits package, including performance-based bonuses, comprehensive health coverage, retirement contributions, wellness days, paid time off, a home office stipend, a professional development budget, a well-being stipend, and considerable paid parental leave.
Requirements
- 8+ years experience as an engineer in machine learning with experience building and shipping machine learning models in production environments for user-facing applications
- Experience with shipping LLMs and multimodal models in user-centric products
- Deep understanding of fundamental machine learning concepts, including deep learning, natural language processing, large language models, and reinforcement learning, with proven expertise in one or more of these areas
- Identify difficult problems that are important for the organization, collaborate across teams and the organization to find the best strategy to resolve
- Consistent track record of technical leadership across multiple teams, shipping AI products showing a high degree of collaboration, flexibility, and respect for different perspectives
- Experience with tooling in the AI ecosystem, that are open source and services in the cloud(such as GCP, AWS, Azure)
- Experience with design of experiments, survey design, and large-scale AB testing
- Experience with responsible AI, transparent algorithms, and putting usersโ needs first
- Youโre pragmatic about how to move things forward in specific timeframes including trade-offs and safeguards when implementing new functionality
- Lead and thrive effectively on a distributed team
Responsibilities
- Take on large open-ended problems and drive them to completion, collaborating with other designers and engineers to make the web a better place
- Identify strategic opportunities where Large Language models improve the experience in Firefox
- Lead the design, development, and integration of innovative AI and machine learning models in Firefox, collaborating cross functionally with product management, full stack engineering and design
- Stay up-to-date with the latest advancements in AI, and inform feature development, and ensure our methodologies remain innovative and relevant
- Lead by example to upskill the team on AI and generate thought leadership
Preferred Qualifications
- Experience with working on on-device machine learning models
- You have previously successfully contributed to an open source project
- Experience with the browser architecture or ecosystem
Benefits
- Generous performance-based bonus plans to all eligible employees - we share in our success as one team
- Rich medical, dental, and vision coverage
- Generous retirement contributions with 100% immediate vesting (regardless of whether you contribute)
- Quarterly all-company wellness days where everyone takes a pause together
- 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. varies by country)