N
Hiring companyNexthire
Verified today
Senior AI/ML Engineer - Computer Vision & Generative AI
Bengaluru, India Full-time Hybrid
About the role
Senior AI/ML Engineer tasked with building production‑grade computer vision and generative AI systems for a no‑code 3D editor platform. Responsibilities include designing object detection, segmentation, and background removal models, fine‑tuning diffusion and GAN pipelines for marketing assets, and deploying scalable inference services on AWS. The role demands deep expertise in PyTorch/TensorFlow, cloud MLOps, and real‑time optimization while driving end‑to‑end ML workflows that improve product image processing and user personalization.
What you’ll do
- Build AI engines for automated image processing, generative content creation, and intelligent workflows
- Develop computer vision models for object detection, segmentation, background removal, and image enhancement
- Fine‑tune generative models for visual and marketing asset creation
- Design and deploy production‑grade ML pipelines on AWS
- Implement model versioning, A/B testing, CI/CD for ML
- Optimize inference pipelines for low latency and high throughput
- Monitor model drift, quality, and performance in production
- Prototype new AI/ML trends and integrate them into the production stack
What you’ll bring
- 5-8+ years experience in AI/ML with strong computer vision background
- Deep expertise in PyTorch/TensorFlow and production ML deployment
- Strong Python skills and ML system design experience
- Cloud experience (AWS/GCP/Azure) and proven record shipping ML products
- Passion for experimenting with new AI models
Nice to have
- Experience in e-commerce/retail imaging or content pipelines
- Background in automation and intelligent workflow systems
- Recommendation/personalization experience
- Familiarity with multimodal models (vision + language)
- Experience with neural rendering or 3D generation
- Open-source contributions or research publications
- Real-time inference optimization
- Strong MLOps understanding
Skills
PyTorch, TensorFlow, Hugging FaceOpenCV, YOLO, Detectron2Stable Diffusion, ControlNet, Diffusersscikit-learn, XGBoostFastAPI, ONNX, TorchScript, TensorRTAWS SageMaker, Lambda, EC2, S3Docker, KubernetesPostgreSQL, Redis, MongoDB, PineconePython, REST, GraphQL, WebSocket
Benefits
- Competitive salary
- Significant stock options (4-year vesting)
- Performance-based bonuses