Mozilla.ai

Machine Learning Engineer

18 April 2024
Apply Now
Deadline date:
£159000 - £245000 / year

Job Description

About Us:

Moz://a.ai is at the forefront of the AI revolution, advocating for a decentralized and open-source approach. Our ambition is to empower developers to craft AI solutions that are both scalable and trustworthy. Through innovation, collaboration, and responsible AI practices, we’re shaping an AI future anchored in user agency, privacy, and transparency.

  • Position: Machine Learning Engineer 
  • Location: Remote (Spain, Portugal, UK) 
  • Type: Full-Time
  • Expected Start Date: Q2 2024

Position Overview: 

We are seeking a skilled Machine Learning Engineer with a strong background in Natural Language Processing (NLP) and Large Language Models (LLMs) to join our team. The ideal candidate will work with foundational OSS language models and specialize them. The candidate will prepare datasets for fine-tuning, evaluate results, and serve the models for production. The main responsibilities of this position are:

  • Develop and implement use cases involving language model fine-tuning and validation.
  • Prepare and preprocess datasets for model fine-tuning and training.
  • Fine-tune OSS foundational models for domain specific tasks.
  • Evaluate model performance using standard metrics and techniques.
  • Design and deploy scalable and efficient NLP pipelines on our cloud infrastructure.
  • Develop APIs for dataset management and model serving.
  • Collaborate with cross-functional teams to integrate NLP functionalities into the main product.
  • 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.
  • Comfortable working on tooling to enable any of the above

Qualifications:

  • Strong background in machine learning, NLP, and deep learning.
  • Proven experience developing and fine-tuning language models, particularly using PyTorch.
  • Experience with large-scale dataset processing and data augmentation techniques.
  • Experience with platforms like Weight & Biases for experiment tracking and VectorDB for embedding storage is plus.
  • Proficiency in Python and familiarity with relevant libraries and tools.
  • Knowledge of APIs and cloud computing platforms (e.g., AWS, GCP), with containerization technologies (e.g., Docker) being a plus.
  • 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.

Please don’t hesitate to get in touch if you have any questions about this role or how you can bring your unique skills to our team.

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.

  1. Purpose-Driven Mission: we are a mission-driven early stage company. If you are passionate about the transformative potential of AI and committed to ensure AI solutions that are trustworthy and responsible.
  2. Innovation & Impact: cutting-edge AI projects that have a real impact on people’s lives.
  3. 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.
  4. 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 (check our comprehensive list of core benefits per location) 
  • 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