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-8 years of experience in development roles.
- 3+ years of hands-on 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 variants, YOLO family, Faster R-CNN, Mask2Former, SAM, and Depth Anything.
- Hands-on experience with generative AI and 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 Python.
- Experience with FastAPI, Flask, Docker, Linux, and Git.
- Familiarity with cloud platforms such as AWS, GCP, Azure, or RunPod.
- 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