Machine Learning Engineer

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?

Submit your application today and join our talented team at Shiprocket.