Job Description
About the Role
We are hiring a Machine Learning Engineer with a strong foundation in computer vision, image
classification, image processing, and prompt-based generative modeling. In this role, you will focus
on building and deploying production-grade ML pipelines that process images at scale, integrate
generative models, and power visual AI products.
Responsibilities
- Build and optimize ML pipelines for image classification, detection, and segmentation tasks.
- Design, train, fine-tune, and deploy deep learning models using CNNs, Vision Transformers, and
diffusion-based models.
- Work with image datasets (structured/unstructured), including preprocessing, augmentation,
normalization, and enhancement techniques.
- Implement and integrate prompt-based generative models (e.g., Stable Diffusion, DALLE, or
ControlNet).
- Collaborate with backend and ...
Ready to Apply?
Take the next step in your AI career. Submit your application to Shiprocket today.
Submit Application