Staff Machine Learning Engineer

AI71 is an applied research team committed to creating helpful and responsible AI agents for knowledge workers. We collaborate with industry partners, leveraging cutting-edge research from the Technology Innovation Institute (TII). Our mission is to bridge the gap between breakthrough AI research and practical real-world applications, building tools that empower individuals, streamline workflows, and enhance knowledge accessibility.

We are seeking a highly motivated and curious Staff Machine Learning Engineer to join our team in Abu Dhabi, UAE (Full Relocation Support Provided). 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 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 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.
• Work with cross-functional teams to understand data and modeling needs.
• Preprocess and clean structured and unstructured data for ML workflows.
• Perform exploratory data analysis and feature engineering.
• Support model training, evaluation, and tuning.
• Stay updated on the 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).

Language Requirements:

• Proficiency in English is expected.

This role offers a unique opportunity for growth within a forward-thinking organization that prioritizes innovation in AI.