Overview
The Role:
We’re hiring a Senior AI/ML Engineer with deep expertise in computer vision, generative AI, and production-grade ML systems. This is a 100% hands-on individual contributor role where you’ll build the AI engines behind our platform—automated image processing, generative content creation, intelligent workflows, and large-scale ML pipelines. You’ll work across computer vision, generative models, automation, and ML infrastructure to deliver production-ready AI systems.
What You’ll Build:
1. Computer Vision & Image Understanding
● Product image analysis, object detection, segmentation
● Automated background removal, image enhancement, preprocessing
● Classification, attribute extraction, and visual search systems
● Quality assessment and edge-case detection models
● Depth estimation and scene understanding from 2D images
● Real-time object detection for AR try-on
● Multi-view image analysis and camera pose estimation
2. Generative AI & Content Creation
● Fine-tune generative models for visual and marketing asset creation
● Text-to-image and image-to-image model pipelines
● AI-generated product descriptions, tags, and metadata
● Work with diffusion models, GANs, transformers
● Texture generation, style transfer, image editing tools
● Synthetic data generation pipelines
● Experiment with the latest foundation models and diffusion techniques
3. Intelligent Automation & ML Systems
● End-to-end automation for large-scale product catalog processing
● Recommendation and personalization models
● Automated workflows for QC, moderation, and validation
● Predictive models for engagement and conversion
● Anomaly detection and platform monitoring
● Continuous learning and self-improving systems
4. Production ML Infrastructure
● Deploy/optimize ML models on AWS
● Build scalable inference pipelines (low latency, high throughput)
● Implement model versioning, A/B testing, CI/CD for ML
● Data pipelines for annotation, augmentation, and quality control
● Optimize models for speed, efficiency, and cost
● Monitoring systems for drift, quality, and performance
● APIs/microservices for ML model serving
5. Research & Innovation
● Explore latest AI/ML trends and cutting-edge models
● Prototype quickly with state-of-the-art models (GPT-4V, Diffusion, SAM, etc.)
● Integrate open-source tools into our production stack
● Run feasibility experiments and contribute to model architecture decisions
● Document learnings and share insights internally
Technical Stack:
1. AI/ML Frameworks
● PyTorch, TensorFlow, Hugging Face
● OpenCV, YOLO, Detectron2
● Stable Diffusion, ControlNet, Diffusers
● scikit-learn, XGBoost
2. Deployment & Infrastructure
● FastAPI, ONNX, TorchScript, TensorRT
● AWS (SageMaker, Lambda, EC2, S3)
● Docker, Kubernetes
● PostgreSQL, Redis, MongoDB, Pinecone
3. Languages & APIs
● Python (primary), JavaScript/Node.js (working knowledge)
● REST, GraphQL, WebSocket
4. Nice to Have (3D/Graphics)
● Understanding of rendering pipelines
● Familiarity with glTF/USDZ
● Experience with Three.js or Unity/Unreal
What We’re Looking For:
Must-Haves
● 5-8+ years’ experience in AI/ML, strong computer vision background
● Deep expertise in PyTorch/TensorFlow
● Production ML deployment experience
● Strong understanding of CNNs, transformers, detection, segmentation
● Hands-on experience with diffusion models or GANs
● Strong Python skills and ML system design
● Cloud experience (AWS/GCP/Azure)
● Proven record of shipping ML products
● Passion for experimenting with new AI models
Location & Work Culture:
● Bengaluru (On-site)
● 6 days a week (5 days office, Saturday WFH)