Data Scientist – Computer Vision

Astek Middle East; Dubai; On-site; AED 250,000 - 400,000; Full time

Job summary

Astek Middle East is seeking an experienced Data Scientist with strong hands-on expertise in computer vision and applied machine learning. You will develop and fine-tune vision-based AI models for object detection, tracking, and segmentation using PyTorch, TensorFlow, and OpenCV.

Responsibilities include curating datasets, training and evaluating models, tracking experiments with MLflow or Weights & Biases, and optimizing for deployment using TensorRT and ONNX.

Qualifications

• Bachelor's or Master's degree in Computer Science, Data Science, Machine Learning, or a related field.
• 3–7 years of hands-on applied Machine Learning experience with a strong focus on Computer Vision.
• Strong Python programming skills with PyTorch or TensorFlow and OpenCV.

Responsibilities

• Develop, train, and fine-tune computer vision models for object detection, object tracking, and image segmentation using PyTorch, TensorFlow, and OpenCV.
• Work hands-on with modern architectures, including YOLO, RF-DETR, and Vision-Language Models (VLMs).
• Curate, preprocess, clean, and augment image and video datasets in collaboration with annotation teams.
• Train, evaluate, and iteratively improve models against performance metrics such as mAP, precision, and recall.
• Track experiments, model versions, and performance results using MLflow or Weights & Biases.
• Optimise models for inference and deployment using TensorRT and ONNX, considering real-world deployment constraints.

Skills

Python programming

Computer Vision

Applied ML

Education

Bachelor's degree in CS/DS/ML

Master's degree in CS/DS/ML

Tools

PyTorch

TensorFlow

OpenCV

TensorRT

ONNX

MLflow

Weights & Biases

Job description

We are looking for an experienced Data Scientist with strong hands-on expertise in Computer Vision and Applied Machine Learning to develop, fine-tune, evaluate, and optimise vision-based AI models. The ideal candidate will have practical experience working with modern object detection, tracking, and segmentation architectures, using frameworks such as PyTorch, TensorFlow, and OpenCV.

You will work with image and video datasets, experiment with state-of-the-art computer vision models, and optimise model performance for real-world deployment requirements.

Key Responsibilities

• Develop, train, and fine-tune computer vision models for object detection, object tracking, and image segmentation using PyTorch, TensorFlow, and OpenCV.
• Work hands-on with modern architectures, including YOLO, RF-DETR, and Vision-Language Models (VLMs), adapting and fine-tuning models for specific use cases.
• Curate, preprocess, clean, and augment image and video datasets in collaboration with annotation teams.
• Train, evaluate, and iteratively improve models against performance metrics, including mean Average Precision (mAP), precision, and recall.
• Track experiments, model versions, and performance results using tools such as MLflow or Weights & Biases.
• Optimise models for inference and deployment using technologies such as TensorRT and ONNX, taking real-world performance and deployment constraints into account.

Requirements

• Bachelor's or Master's degree in Computer Science, Data Science, Machine Learning, or a related field.
• 3–7 years of hands-on applied Machine Learning experience, with a strong focus on Computer Vision.
• Strong Python programming skills and practical experience with PyTorch or TensorFlow, together with OpenCV.
• Demonstrated experience developing or fine-tuning models for object detection, tracking, and segmentation.
• Practical knowledge of computer vision architectures such as YOLO; experience with RF-DETR is relevant.
• Familiarity with model evaluation metrics, dataset preparation, data augmentation, and iterative model improvement.
• Experience with experiment-tracking tools such as MLflow or Weights & Biases.
• Familiarity with model optimisation and inference technologies such as ONNX and TensorRT.

Preferred:

Hands-on experience fine-tuning Vision-Language Models (VLMs).

Important Application Criteria

Candidates must meet the relevant technical experience requirements and have practical Computer Vision experience.