AI/ML Engineer – Computer Vision & Generative AI

Company: Lotus Prime Digital Solutions
Location: Whitefield, Bangalore, Karnataka
Job Type: Full-time, Permanent
Work Mode: Work From Office (WFO)
Experience: 2–4 Years
Industry: Artificial Intelligence, Machine Learning, Computer Vision & Software Development

About the Company

Lotus Prime Digital Solutions is a technology-driven company specializing in Artificial Intelligence, Digital Transformation, Software Development, AI-powered Applications and Innovative Digital Solutions.

We are developing advanced AI-based products, including Virtual Try-On Systems, AI Digital Standees, AI Hairstyle Simulation, Jewellery Try-On, AI-powered Kiosks and Computer Vision Applications.

We are looking for a talented and passionate AI/ML Engineer specializing in Computer Vision and Generative AI to join our development team.

The selected candidate will work on developing, fine-tuning, optimizing and deploying AI models for real-world commercial applications.

Key Responsibilities

1. AI Model Development

- Develop, train, fine-tune and optimize AI/ML models.
- Research and implement open-source AI models for commercial applications.
- Work with deep learning architectures, including CNNs, Transformers and diffusion models.
- Develop AI-powered image generation, image transformation and image processing solutions.
- Evaluate AI model performance, accuracy, inference speed and resource utilization.

2. Computer Vision Development

- Develop face detection, facial landmark detection and facial alignment systems.
- Implement human body segmentation, pose estimation and body landmark detection.
- Develop image masking, object detection and image enhancement algorithms.
- Work on realistic image transformations while preserving facial identity and body proportions.
- Improve the accuracy and visual quality of AI-generated images.

3. Virtual Try-On AI Development

- Develop AI-powered virtual clothing try-on applications.
- Work on realistic saree, dress and garment fitting technologies.
- Develop jewellery try-on solutions for necklaces, earrings and other accessories.
- Implement hairstyle, beard and makeup simulation technologies.
- Improve garment positioning, jewellery alignment, image consistency and realistic appearance.




- Research methods for improving virtual fitting accuracy and preserving original product details.

4. Generative AI & Deep Learning

- Work with Stable Diffusion, diffusion-based image generation and other generative AI architectures.
- Implement and fine-tune pretrained AI models using PyTorch.
- Explore LoRA, ControlNet and other model adaptation techniques.
- Work with image-to-image generation, inpainting and image enhancement.
- Optimize generative AI pipelines for faster and more cost-effective inference.

5. AI API Development & Integration

- Develop REST APIs for AI model inference using Python and FastAPI.
- Integrate AI models with web applications, mobile applications and digital kiosks.
- Deploy AI models on GPU-based servers and cloud infrastructure.
- Optimize AI processing costs, inference time and scalability.
- Collaborate with frontend and backend developers to integrate AI functionality.

6. Research & Development

- Evaluate emerging AI models and open-source frameworks.
- Conduct proof-of-concept development for new AI products.
- Identify opportunities to replace expensive third-party AI APIs with in-house solutions.
- Improve existing AI systems through testing, optimization and model fine-tuning.
- Maintain technical documentation for models, APIs and deployment processes.

Required Technical Skills

Programming Languages

- Strong proficiency in Python.
- Experience with NumPy, OpenCV and image processing libraries.
- Basic knowledge of JavaScript or TypeScript is an advantage.

AI & Machine Learning

- PyTorch
- Deep Learning
- Computer Vision
- Generative AI
- Diffusion Models
- Image Processing
- Model Training and Fine-Tuning
- Hugging Face Transformers and Diffusers

Computer Vision Technologies

- OpenCV
- MediaPipe
- YOLO
- Image Segmentation
- Human Pose Estimation
- Facial Landmark Detection
- Object Detection and Tracking

Backend & Deployment

- FastAPI or Flask
- REST API Development
- Docker
- Git and GitHub
- GPU Computing and CUDA




- Cloud GPU Deployment

Preferred Skills

Candidates with experience in the following areas will receive additional consideration:

- AI Virtual Try-On technologies
- Virtual Clothing Fitting
- Jewellery Virtual Try-On
- AI Hairstyle Generation
- AI Makeup Simulation
- Stable Diffusion and ControlNet
- LoRA Fine-Tuning
- Human Parsing and Body Segmentation
- ONNX Runtime or TensorRT
- GPU Model Optimization
- Self-hosted AI Models
- Commercial AI API Development

Educational Qualifications

Bachelor's or Master's degree in:

- Computer Science
- Artificial Intelligence
- Machine Learning
- Data Science
- Information Technology
- Electronics or related engineering disciplines

Candidates with strong practical AI development experience and demonstrable projects will also be considered.

Candidate Requirements

- Minimum 2 years of relevant AI/ML development experience preferred.
- Strong problem-solving and analytical skills.
- Ability to independently research and implement AI models.
- Practical experience with computer vision and image-based AI applications.
- Understanding of GPU infrastructure and model optimization.
- Ability to develop production-ready AI solutions.
- Willingness to work on innovative AI products and emerging technologies.
- Good communication and teamwork skills.

What We Offer

- Opportunity to develop innovative AI products.
- Hands-on experience with commercial AI applications.
- Exposure to computer vision, generative AI and deep learning technologies.
- Opportunity to work on proprietary AI model development.
- Collaborative development environment.
- Long-term growth opportunities opportunities.

Application Requirements

Interested candidates should submit:

- Updated resume.
- GitHub profile or AI project portfolio, if available.
- Details of AI/ML models previously developed or fine-tuned.
- Experience with computer vision or generative AI projects.
- Current CTC, expected CTC and notice period.

Work Location: Whitefield, Bangalore
Company: Lotus Prime Digital Solutions
Website: lotusprimedigitalsolutions.com
Contact: +91-XXXXXXXXXX

Join us in building the next generation of AI-powered products and intelligent digital solutions.

Pay: ₹28,000.00 - ₹50,000.00 per month

Work Location: In person

📌 AI/ML Engineer – Computer Vision & Generative AI (Whitefield)
🏢 Lotus Prime Digital Solutions
📍 Whitefield