AI ML Platform Jobs in San Francisco, CA

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 AI Platform team is responsible for building the foundational infrastructure that powers intelligent experiences across Mozilla products. This includes model training pipelines, high-throughput inference services, GPU orchestration, and secure, privacy-respecting AI systems that operate reliably at global scale.

We're looking for a Machine Learning Engineer with a strong platform mindset to help design, build, and operate Mozilla's AI platform. In this role, you'll work at the intersection of machine learning, distributed systems, and production infrastructure—ensuring that models can be trained, deployed, and served efficiently, securely, and at scale. You will collaborate closely with product, infrastructure, and security teams to enable fast iteration while meeting strict performance and privacy requirements.

What You'll Do:

• Design, build, and operate core AI platform components used to train, deploy, and serve machine learning models in production environments.
• Own model serving and inference workflows end-to-end, driving improvements in reliability, scalability, performance, and operational excellence.
• Lead efforts to optimize inference systems for throughput, latency, and cost efficiency across CPU and GPU workloads.
• Design and manage GPU-based inference and training workloads, including performance tuning, capacity planning, and resource utilization optimization.
• Own and improve critical parts of the model lifecycle, including packaging, versioning, testing strategies, validation, and deployment automation.
• Implement and evolve observability practices (metrics, logging, tracing, alerting) to improve visibility and operational resilience of ML services and pipelines.
• Partner closely with product, infrastructure, security, and data teams to design scalable platform capabilities that enable AI-powered features.
• Contribute to technical design discussions, propose architectural improvements, and mentor junior engineers through code reviews and knowledge sharing.
• Participate in and help improve operational processes, including incident response, on-call rotations, and post-incident reviews.

What You'll Bring:

• Bachelor's degree with 4–6 years of relevant industry experience, or Master's degree with significant hands-on experience building and operating production ML systems, or work experience equivalent
• Strong experience developing in Python for machine learning systems, backend services, or distributed data processing.
• Proven experience deploying and operating ML workloads in cloud environments, including production-grade infrastructure.
• Solid understanding of model serving architectures, inference pipelines, and performance tradeoffs (latency, throughput, cost, scaling strategies).
• Hands-on experience working with GPU-based workloads and accelerated computing in production settings.
• Experience designing CI/CD pipelines and development workflows that support reliable ML system deployment.
• Ability to independently scope and drive technical initiatives while balancing product and operational priorities.
• Strong problem-solving skills and the ability to debug performance and reliability issues in distributed systems.
• Clear and effective communication skills, with experience collaborating across engineering, product, and infrastructure teams.

Bonus Skills:

• Experience implementing inference optimization strategies such as batching, quantization, compilation, model conversion, or hardware-specific tuning.
• Familiarity with containerization and orchestration systems (e.g., Docker, Kubernetes) in production environments.
• Experience designing observability systems for distributed services, including metrics strategy and performance profiling.
• Exposure to privacy-preserving ML techniques, security best practices, or responsible AI system design.
• Contributions to open-source ML infrastructure projects or leadership in building reusable internal ML tooling.

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: D

#LI-REMOTE

Req ID: R3074