AI/ML Engineer

Job Title : AI/ML Engineer Location: On-Site Employment Type: Full-time Experience Level: Mid-Senior (2-3+ years experience) We are looking for a Lead AI/ML Engineer to spearhead the development of our advanced generative AI platform designed to transform product images into stunning lifestyle visuals. You will be responsible for building and optimizing cutting-edge AI models tailored specifically for furniture and home décor imagery. Responsibilities: * Develop, fine-tune, and deploy custom generative AI models (Stable Diffusion, ControlNet, etc.) tailored to product and interior imagery. * Design robust ML pipelines for automated image segmentation, scene generation, and image editing (inpainting, recoloring). * Conduct research and experimentation to continually improve image quality, realism, and performance. * Oversee AI infrastructure setup and manage GPU-based model inference workloads. * Stay up-to-date with state-of-the-art developments in generative AI and computer vision. Required Skills & Experience: * Strong proficiency in Python and deep-learning frameworks like PyTorch or TensorFlow. * Hands-on experience with generative models, especially Stable Diffusion, GANs, ControlNet, or similar. * Experience training and fine-tuning custom models on domain-specific datasets (image-based models strongly preferred). * Practical experience with segmentation models, object detection, and inpainting techniques. * Familiarity with HuggingFace, Diffusers, or similar model frameworks. * Experience deploying ML models to production environments using cloud infrastructure (AWS, Azure, GCP). * Strong mathematical foundation and familiarity with computer vision/image processing. Preferred Qualifications: * Previous experience building generative AI products (image, video). * Familiarity with containerization technologies (Docker, Kubernetes). * Background in e-commerce, furniture, fashion-tech, or similar industries. Personal Attributes: * Curious, proactive, and self-driven. * Excellent communication skills. * Team-oriented with leadership potential. Job Type: Full-time Benefits: Paid sick time Paid time off Provident Fund Schedule: Day shift Work Location: In person Application Deadline: 30/06/2025 Expected Start Date: 06/10/2025