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.

Together, these initiatives are shaping an AI future anchored in user agency, trustworthiness, and transparency.

Position Overview

We are seeking a Machine Learning Engineer with a strong background in developing products with ML models at their core - areas like recommendation systems, scaled detection, feature extraction, content modeling, agents, natural language interfaces, search, etc., to join our dynamic team. The ideal candidate will be working on a 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 development, management, and/or 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.
• Collaborate on the implementation of 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, (e.g., PyTorch) is a plus.
• Experience with large-scale dataset processing and data augmentation techniques.
• Experience with managing experiments and models (e.g., platforms like Weights and Biases, MLFlow, KubeFlow).
• Strong proficiency in Python and familiarity with relevant libraries and tools.
• Comfortable working in the cloud - containerization, cloud resource management, etc.
• 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 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.

• Purpose-Driven Mission: We are a mission-driven early-stage company. If you are passionate about the transformative potential of AI and committed to ensuring AI solutions that are trustworthy and responsible.
• Innovation & Impact: Cutting-edge AI projects that have a real impact on people's lives.
• Collaborative Culture: Our team is distributed across different countries, fostering a collaborative and inclusive culture where everyone's input is valued. We make sure to meet several times a year to work together in a place in the world defined in advance.
• Remote Work: We are a 100% remote team, distributed around the world. Since we do not have offices in all locations, we partner with an Employer of Record.

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

Compensation, Benefits, and Perks

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