Mozilla

Mozilla

Senior Machine Learning Engineer, AI Platform

Remote US · Senior

Sponsorship not specified$163k-$218kDetected 27 days ago
PythonDistributed SystemsBackend DevelopmentCode ReviewCloud PlatformsDockerKubernetesCI/CDMachine LearningMLOpsIncident ResponseLeadershipCommunicationProblem Solving

About the role

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

Responsibilities

  • 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.
  • 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.
  • Exposure to privacy-preserving ML techniques, security best practices, or responsible AI system design.

Requirements

  • 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
  • Proven experience deploying and operating ML workloads in cloud environments, including production-grade infrastructure.
  • Hands-on experience working with GPU-based workloads and accelerated computing in production settings.
  • 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.

Nice to have

  • Experience implementing inference optimization strategies such as batching, quantization, compilation, model conversion, or hardware-specific tuning.

Skills

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

Compensation

  • US Tier 1 Locations
  • $163,000 - $218,000 USD
  • US Tier 2 Locations
  • $150,000 - $200,000 USD
  • US Tier 3 Locations
  • $139,000 - $185,000 USD
  • Annual professional development budget

Benefits

  • Employee referral bonus program
  • Other benefits (life/AD&D, disability, EAP, etc. - varies by country)
  • Commitment to diversity, equity, inclusion, and belonging

Company info

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

Visa & Work Authorization

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

This listing is sourced directly from Mozilla's careers page and normalized into a canonical job model.