Remote Machine Learning Engineer

About Us

Mozilla.ai is at the forefront of the AI revolution, advocating for a truly open-source approach. Our ambition is to empower developers to craft AI solutions that are both scalable and trustworthy. Through Lumigator, our model selection platform, we provide tools for evaluating and selecting the most appropriate models for various use cases, ensuring robustness and reliability. Additionally, our open source AI Hub fosters collaboration and innovation in the open-source community, bringing like-minded developers and organizations together to promote responsible AI practices.

Position Overview

We are seeking a Machine Learning Engineer with a strong background in developing products with ML models at their core. Areas of focus include recommendation systems, scaled detection, feature extraction, content modeling, agents, natural language interfaces, and search. The ideal candidate will join our dynamic product engineering team and contribute to our open-source LLM evaluation platform, focusing on model validation, selection, and deployment.

Main Responsibilities

• Develop tools to streamline the development, management, and evaluation of models.
• Collaborate on the design and build of end-to-end machine learning pipelines on cloud infrastructure.
• Prepare and preprocess datasets for model evaluation.
• Run and manage experiments related to model evaluation and training.
• Evaluate model performance using standard metrics and techniques.
• Collaborate with other teams, including product management and platform engineering, to ship features.
• Implement MLOps best practices to ensure smooth model deployment, monitoring, and maintenance.
• Engage with internal and external stakeholders, translating complex technical details into clear insights.
• Contribute to the product engineering lifecycle, from ideation to deployment and maintenance of new features.

Qualifications

• Strong background in machine learning and software engineering.
• Comfortable with open source development work.
• Experience with deep learning frameworks, such as PyTorch, is a plus.
• Experience with large-scale dataset processing and data augmentation techniques.
• Experience with managing experiments and models using platforms like Weights and Biases, MLFlow, or KubeFlow.
• Strong proficiency in Python and familiarity with relevant libraries and tools.
• Comfortable working in the cloud, including containerization and cloud resource management.
• Excellent problem-solving skills and the ability to work both independently and as part of a team.
• Effective communication skills, including the ability to translate technical concepts to non-technical stakeholders.
• A demonstrated track record of delivering high-quality, scalable solutions in a fast-paced environment.

Why Join Us

We are more than just a company; we are a community of like-minded individuals driven by a shared passion for creating positive change in society through AI solutions. We offer a purpose-driven mission, innovation and impact through cutting-edge AI projects, and a collaborative culture. Our team is distributed across different countries, fostering a collaborative and inclusive culture where everyone's input is valued. We are a 100% remote team, distributed around the world.

Compensation, Benefits, and Perks

• Premium package featuring core benefits tailored to your country of residence, including health insurance and retirement plans.
• 25 days per year of paid time off.
• Generous performance-based bonus plans for all regular employees.
• One-time home office stipend of 1,000 USD.
• Annual professional development budget.
• Annual well-being stipend of 3,500 USD.

We are committed to building a diverse and inclusive team. We encourage applications from individuals of all backgrounds, beliefs, and identities.