Staff Machine Learning Engineer, Gen AI

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, fine-tune and evaluate LLMs and GenAI 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 Bring…

• 6+ years experience as a Machine Learning engineer with working knowledge of Python and experience with deep learning frameworks such as Pytorch, TensorFlow, JAX or MXNet.
• Experience with fine-tuning and evaluating LLM models.
• Experience working in machine learning tooling and infrastructure for training, deploying and inference of models. 
• Analytical ability to apply and solve sophisticated machine learning problems spanning web and mobile experiences. 
• Strong problem-solving skills and ability to communicate complex concepts to technical and non-technical stakeholders.
• Experience with design of experiments, survey design, and large-scale AB testing.
• Experience working collaboratively with product managers, project managers, and/or other non-engineering teams.
• Good written and verbal English communication skills and have participated effectively on a distributed team.
• Commitment to our values:
  • Welcoming differences
  • Being relationship-minded
  • Practicing responsible participation
  • Having grit