Staff Machine Learning Engineer, Gen AI

6 Months ago • 6 Years + • Full Stack Development

Job Summary

Job Description

Mozilla seeks a Staff Machine Learning Engineer to lead Generative AI solutions within Firefox. You'll design, build, and integrate LLMs and small models, ensuring safe and reliable experiences. Expertise in ML tooling, distributed systems, and LLM APIs is essential.
Must have:
  • Machine Learning Engineer
  • LLM API Experience
  • Distributed Systems
  • Production ML Tooling
Good to have:
  • Cloud Provider LLMs
  • Open Source Tooling
  • Browser Extensions
  • Responsible AI
Perks:
  • Performance-based Bonus
  • Generous Retirement

Job Details

Why 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. Now, with more than 225 million people around the world using our products each month, we’re shaping the next 25 years of technology and helping to reclaim an internet built for people, not companies. 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 Mozilla Corporation is wholly owned by the non-profit 501(c) Mozilla Foundation. This means we aren’t beholden to any shareholders — only to our mission. Along with thousands of volunteer contributors and collaborators all over the world, Mozillians design, build and distribute open-source software that enables people to enjoy the internet on their terms.

About this team and role:

The Firefox team is a community of engineers who care deeply about delivering the fastest, friendliest, most usable browser possible. We are responsible for making the things you see in the browser work securely, quickly, and well! 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. 

What is a Staff Engineer at Mozilla? 

A Staff Engineer is the next level from a Senior Engineer. At Mozilla this role can vary but typically a Staff Engineer is a domain expert who leads others within a single team to tackle multi-month projects. Tasks that may be initially ambiguous or require significant planning and require them to influence or direct the work of several engineers. They sequence deliverables and manage risks in their team's projects, provide feedback on our strategy and goals affecting the team, and turn our strategy into action for their team members. Staff Engineers mentor others by stewarding some responsibilities to more junior and senior engineers so they can take on new ones. They collaborate with management on building team consensus and providing direction. Staff Engineers identify gaps and opportunities for improvement to enable a culture of inclusion and allyship, at all levels of the organization.

What you’ll do:

  • 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

What you'll bring:

  • 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.
  • Commitment to our values:
    • Welcoming differences
    • Being relationship-minded
    • Practicing responsible participation
    • Having grit

Bonus Points

  • Knowledge of and experience with using LLM solutions across different cloud providers
  • You have previously successfully contributed to an open source project.
  • Track record of not only building one, but multiple projects deployed with an active user base.
  • Experience with responsible AI, transparent algorithms, and putting users’ needs first.
  • Excellent communication skills, both in written and presentation form. You have the ability to quickly distill sophisticated topics into concepts that are tailored for your target audience.
  • You are a lifelong learner, and continue to refine and improve your skills when you see an opportunity to do so.
  • Experience with the browser extensions architecture or ecosystem.
  • Experience working in open source environments.

What you’ll get:

  • 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)

About Mozilla 

Mozilla exists to build the Internet as a public resource accessible to all because we believe that open and free is better than closed and controlled. When you work at Mozilla, you give yourself a chance to make a difference in the lives of Web users everywhere. And you give us a chance to make a difference in your life every single day. Join us to work on the Web as the platform and help create more opportunity and innovation for everyone online.

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.

We will 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. Please contact us at hiringaccommodation@mozilla.com to request accommodation.

We are an equal opportunity employer. We do not discriminate on the basis of race (including hairstyle and texture), religion (including religious grooming and dress practices), gender, gender identity, gender expression, color, national origin, pregnancy, ancestry, domestic partner status, disability, sexual orientation, age, genetic predisposition, medical condition, marital status, citizenship status, military or veteran status, or any other basis covered by applicable laws.  Mozilla will not tolerate discrimination or harassment based on any of these characteristics or any other unlawful behavior, conduct, or purpose.

Group: C

#LI-REMOTE

Req ID: R2593

Similar Jobs

Varonis  - Python Developer

Varonis

Herzliya, Tel Aviv District, Israel (Hybrid)
4 Months ago
Google - Staff Software Engineer, Workspace Monetization, Mobile

Google

Sunnyvale, California, United States (On-Site)
2 Days ago
ByteDance - Algorithm Intern (Video Codec - Realtime Codec Optimizations - Multimedia Streaming) - 2025 Summer (PhD)

ByteDance

San Diego, California, United States (On-Site)
4 Months ago
Google - Staff Software Engineer, Google Pay Vouchers and Growth

Google

Bengaluru, Karnataka, India (On-Site)
1 Week ago
Tencent - Speech Synthesis Intern

Tencent

London, England, United Kingdom (On-Site)
2 Months ago
ByteDance - Software Engineer

ByteDance

San Jose, California, United States (On-Site)
2 Weeks ago
Google - Software Engineer II, Android Wear Google Messages

Google

Kraków, Lesser Poland Voivodeship, Poland (On-Site)
2 Weeks ago
N-iX - Senior Full Stack Engineer (.NET, with focus on React)

N-iX

Colombia (Remote)
2 Weeks ago
Google - Senior Customer Solutions Engineer

Google

São Paulo, State Of São Paulo, Brazil (On-Site)
2 Weeks ago
Cyara - Professional Services Consultant

Cyara

Hyderabad, Telangana, India (Hybrid)
5 Months ago

Get notifed when new similar jobs are uploaded

Similar Skill Jobs

Google - Software Engineer, Kernel, ChromeOS

Google

Bengaluru, Karnataka, India (On-Site)
2 Weeks ago
Electronic Arts - Software Developer II - AdTech

Electronic Arts

Vancouver, British Columbia, Canada (Hybrid)
2 Weeks ago
social point  - Senior Software Engineer (GameOps Tools)

social point

Barcelona, Catalonia, Spain (Hybrid)
17 Hours ago
NVIDIA - Senior Software Video Engineer

NVIDIA

Ra'anana, Center District, Israel (On-Site)
1 Month ago
Google - Software Engineer III, Infrastructure, Google Cloud Platforms

Google

Kirkland, Washington, United States (On-Site)
5 Months ago
NVIDIA - Senior Software Engineer, Place and Route Tools

NVIDIA

Austin, Texas, United States (On-Site)
2 Weeks ago
Google - Staff Software Engineer Technical Lead, Storage Transfer Appliances

Google

Sunnyvale, California, United States (On-Site)
2 Weeks ago
Google - Software Engineer, Site Reliability Engineering, Campus

Google

Sydney, New South Wales, Australia (On-Site)
2 Days ago
Meta - Software Engineer, Systems ML - SW/HW Co-design

Meta

Redmond, Washington, United States (On-Site)
5 Months ago
Magna International - ADAS Feature Owner

Magna International

Bengaluru, Karnataka, India (On-Site)
7 Months ago

Get notifed when new similar jobs are uploaded

Jobs in Finland

Supercell - IT Technical Producer

Supercell

Helsinki, Uusimaa, Finland (On-Site)
1 Day ago
Remedy Entertainment Plc - Principal Hard Surface Artist

Remedy Entertainment Plc

Helsinki, Uusimaa, Finland (On-Site)
5 Months ago
Metacore - UI/UX Lead

Metacore

Helsinki, Uusimaa, Finland (Hybrid)
2 Months ago
Keywords Studios - Expert Subtitle Translator/QCer: English to Finnish

Keywords Studios

Finland (Remote)
2 Weeks ago
Remedy Entertainment Plc - Senior Audio Designer

Remedy Entertainment Plc

Helsinki, Uusimaa, Finland (On-Site)
1 Month ago
Supercell - Senior Game Tester, New Game: mo.co

Supercell

Helsinki, Uusimaa, Finland (On-Site)
5 Months ago
Supercell - Product Marketing, Clash Royale

Supercell

Helsinki, Uusimaa, Finland (On-Site)
5 Months ago
Netflix - Senior Frontend Developer

Netflix

Helsinki, Uusimaa, Finland (Hybrid)
2 Weeks ago
Supercell - Senior Game Programmer

Supercell

Helsinki, Uusimaa, Finland (On-Site)
1 Day ago
Zaibatsu Interactive - Internship

Zaibatsu Interactive

Jyväskylä, Central Finland, Finland (On-Site)
2 Years ago

Get notifed when new similar jobs are uploaded

Full Stack Development Jobs

ByteDance - Software Engineer - Low-code Platform

ByteDance

Singapore (On-Site)
2 Months ago
Epic Games - Senior Software Engineer, Developer Relations (UE China)

Epic Games

Shanghai, Shanghai, China (On-Site)
2 Weeks ago
Google - Software Engineer III, Infrastructure, Google Cloud Security and Privacy

Google

Kirkland, Washington, United States (On-Site)
4 Months ago
Google - Software Engineer, Metrics and Business Intelligence

Google

Mexico City, Mexico City, Mexico (On-Site)
2 Days ago
Warner Bros Games - Senior Software Engineer - Roku

Warner Bros Games

Bengaluru, Karnataka, India (Hybrid)
2 Months ago
Ness Digital - Senior .NET Full-Stack Engineer

Ness Digital

Timișoara, Timiș, Romania (Remote)
2 Months ago
Rockstar Games - Senior Full Stack Engineer (C#/React)

Rockstar Games

New York, New York, United States (On-Site)
11 Months ago
Google - Software Engineer III, YouTube

Google

San Bruno, California, United States (On-Site)
2 Weeks ago
Electronic Arts - Experienced Online Software Engineer - American Football Live

Electronic Arts

Orlando, Florida, United States (Hybrid)
3 Weeks ago
Aristocrat Gaming - Senior Engineer I

Aristocrat Gaming

Noida, Uttar Pradesh, India (Hybrid)
1 Week ago

Get notifed when new similar jobs are uploaded