Staff Machine Learning Engineer

Job Title: Staff Machine Learning Engineer

Location: Abu Dhabi, UAE (Full Relocation Support Provided). Only apply if you are interested in relocation to UAE.

About Us

AI71 is an applied research team dedicated to creating helpful and responsible AI agents for knowledge workers. Working closely with our industry partners, our cross-functional teams of AI experts build products grounded in the cutting-edge research of our colleagues from the Technology Innovation Institute (TII).

We are passionate about bridging the gap between breakthrough AI research and practical real-world applications — building tools that empower people, streamline complex workflows, and make knowledge more accessible.

Job Summary

We are seeking a highly motivated and curious Staff Machine Learning Engineer to join our team. In this role, you will assist in developing, testing, and deploying machine learning models under the guidance of senior engineers and data scientists. This is a great opportunity for someone looking to kickstart their career in machine learning and contribute to real-world projects.

What You'll Do:

• Analyze large and complex data sets to extract meaningful insights and inform data-driven decision-making.
• Develop, train, and deploy predictive models to enhance the capabilities of our AI solutions.
• Collaborate with cross-functional teams to understand business objectives and translate them into actionable data science tasks.
• Continuously evaluate and improve existing models based on real-world performance and evolving business needs.
• Implement and maintain data preprocessing pipelines to ensure the quality and reliability of input data.
• Stay abreast of advancements in machine learning and data science techniques, applying them to improve model accuracy.
• Communicate findings and insights effectively to both technical and non-technical stakeholders.

Responsibilities

• Develop, train, and deploy predictive models to enhance the capabilities of our AI solutions.
• Work on and apply cutting-edge AI technologies to business contexts, e.g. Retrieval-Augmented Generation (RAG).
• Continuously evaluate and improve existing models based on real-world performance and evolving business needs.
• Implement and maintain data preprocessing pipelines to ensure the quality and reliability of input data.
• Stay abreast of advancements in machine learning and data science techniques, applying them to improve model accuracy.
• Communicate findings and insights effectively to both technical and non-technical stakeholders
• Collaborate with cross-functional teams to understand data and modeling needs
• Preprocess and clean structured and unstructured data for use in ML workflows
• Perform exploratory data analysis and feature engineering
• Support model training, evaluation, and tuning
• Stay updated on latest ML research and tools

Requirements

• Master’s degree or higher in Computer Science, Machine Learning, Engineering, Mathematics, or a related field
• Foundational knowledge of machine learning algorithms and principles
• Proficiency in Python and common ML libraries (e.g., scikit-learn, pandas, NumPy, TensorFlow or PyTorch)
• Basic experience with data processing, model training, and evaluation
• Understanding of version control tools like Git
• Strong analytical and problem-solving skills
• Eagerness to learn and take on new challenges

Preferred Qualifications

• Familiarity with cloud platforms (AWS, GCP, or Azure)
• Exposure to MLOps concepts or frameworks (e.g., MLflow, Docker)