AI Engineer

Role Overview

We are seeking a highly skilled AI Engineer/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, including image understanding, segmentation, object detection, and image generation 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 machine learning, deep learning, and computer vision.
• Proven experience developing, optimizing, and deploying production-grade AI solutions.
• Strong expertise in models such as RF-DETR, YOLO, Mask2Former, and SAM.
• Hands-on experience with Stable Diffusion XL (SDXL), ControlNet, and Hugging Face Diffusers.
• Advanced proficiency in PyTorch and Python.
• Experience with FastAPI, Flask, Docker, and Linux.
• Familiarity with cloud platforms like AWS, GCP, or Azure.
• Knowledge of MLOps, CI/CD, and OpenCV.

Skills

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

Benefits

• Flexible schedule
• Leave encashment
• Provident Fund