Machine Learning Engineer — Computer Vision & Generative AI

Role Overview

We are looking for a hands-on Machine Learning Engineer to join our core technology team and work on an early-stage Computer Vision & Generative AI product. The role involves building ML pipelines, preparing and managing datasets, training and fine-tuning models, and developing reliable computer-vision solutions for real-world applications. This is an ownership-driven role where you will work closely with the founding and engineering team.

Responsibilities

• Build and improve Computer Vision and Deep Learning pipelines
• Prepare, clean, structure and annotate ML datasets
• Develop image-processing and computer-vision solutions
• Train, fine-tune and evaluate deep-learning models
• Work with Generative AI and image-generation models
• Experiment with different model architectures and approaches
• Optimize models for efficient GPU inference
• Build and evaluate working ML prototypes
• Integrate trained models into the existing application stack
• Monitor model performance and continuously improve results

Requirements

• 2–4 years of relevant professional or strong project experience
• B.Tech / M.Tech / MS in Computer Science, AI, ML or a related field preferred
• Strong analytical and problem-solving skills
• Strong Python programming
• Strong hands-on experience with PyTorch
• Solid Machine Learning & Deep Learning fundamentals
• Strong Computer Vision & OpenCV experience
• Experience with image processing, segmentation and object detection
• Experience with human pose estimation / human parsing
• Experience with ML dataset preparation, cleaning and annotation
• Experience with model training, fine-tuning and evaluation
• Experience with Generative AI / Diffusion Models
• Practical experience with GPU / CUDA

Skills

• Python
• PyTorch
• OpenCV
• CUDA
• Generative AI

Benefits

• Provident Fund
• Work from home