Data Scientist

Role Overview

We are seeking a highly skilled Data Scientist with strong expertise in Computer Vision and Generative AI to join our AI team. The ideal candidate will have hands-on experience developing, fine-tuning, and deploying state-of-the-art vision and diffusion models for real-world applications. You will work on advanced image understanding, segmentation, object detection, depth estimation, image generation, and image editing systems.

Responsibilities

• Design, train, fine-tune, and deploy computer vision and generative AI models.
• Develop solutions for object detection, segmentation, depth estimation, image inpainting, and virtual staging applications.
• Build and optimize end-to-end pipelines for image understanding and image generation tasks.
• Evaluate model performance using appropriate metrics and implement improvements.
• Create and maintain data annotation, training, validation, and testing workflows.
• Work closely with engineering teams to productionize AI models and services.
• Research and implement the latest advancements in computer vision, diffusion models, and multimodal AI systems.
• Optimize models for inference speed, memory consumption, and scalability.
• Develop robust APIs and model-serving solutions for production environments.
• Document experiments, model architectures, and deployment processes.

Requirements

• 3-7 years of experience in Machine Learning, Deep Learning, Computer Vision, and Generative AI.
• Proven experience developing, optimizing, and deploying production-grade AI solutions.
• Strong expertise in computer vision models including RF-DETR, DETR, YOLO, Faster R-CNN, Mask2Former, and SAM.
• Hands-on experience with diffusion models such as Stable Diffusion XL (SDXL), ControlNet, and Hugging Face Diffusers.
• Strong understanding of CNNs, Transformers, and Vision Transformers (ViTs).
• Advanced proficiency in PyTorch and model training/fine-tuning.
• Strong Python programming skills with experience in FastAPI or Flask.
• Experience with Docker, Linux, and Git.
• Familiarity with cloud platforms such as AWS, GCP, or Azure.
• Experience with dataset tools like CVAT, Label Studio, or Roboflow.

Skills

• PyTorch
• Python
• Computer Vision
• Generative AI
• Docker

Benefits

• Flexible schedule
• Leave encashment
• Provident Fund