Machine Learning Engineer

1 week ago

Gurugram, India Shiprocket Full time

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 product teams to deploy real-time or batch inference systems (using Docker, TorchServe, TensorRT, etc.).

- Optimize model performance for speed, accuracy, and size (quantization, pruning, ONNX

conversion, etc.).

- Ensure robust versioning, reproducibility, and monitoring of models in production.


Required Skills

- 2-4 years of experience building and deploying ML models in production environments.

- Strong proficiency in Python and deep learning frameworks like PyTorch or TensorFlow.

- Hands-on experience with CNNs, ViTs, UNets, or other architectures relevant to image-based

tasks.

- Experience with prompt-based image generation models (e.g., Stable Diffusion, Midjourney APIs,

DALLE, or open-source alternatives).

- Familiarity with OpenCV, albumentations, or similar libraries for image processing.

- Ability to train and evaluate models on large datasets with proper tracking (e.g., using MLflow or

Weights & Biases).

- Experience with model optimization tools (ONNX, TensorRT, quantization).

- Comfortable working with GPU-based environments and optimizing training/inference

performance.


Nice to Have

- Experience with ControlNet, LoRA, or DreamBooth for custom generative image tuning.

- Familiarity with deployment using TorchServe, FastAPI, or Triton Inference Server.

- Knowledge of cloud infrastructure (e.g., AWS Sagemaker, GCP AI Platform) for scalable

training/inference.

- Basic understanding of CI/CD pipelines for ML (MLOps practices).


What We Offer

- Opportunity to work on cutting-edge generative and visual AI problems.

- Collaborative and engineering-driven culture.

- Access to high-performance GPUs and scalable compute resources.

  • Machine Learning Engineer

    2 weeks ago

    Gurugram, Gurugram, India IndusInd Bank Full time

    Job Description About the Digital Business Unit at IndusInd: - The mandate of the Digital Business Unit at IndusInd Bank is as follows: - Building customer centric products with human centered design principles for retail Individual and micro, small and medium enterprise (MSME) customer segments - Build innovative products and propositions backed with...

  • Machine Learning Engineer

    3 weeks ago

    Gurugram, India IndusInd Bank Full time

    About the Digital Business Unit at IndusInd: The mandate of the Digital Business Unit at IndusInd Bank is as follows: Building customer centric products with human centered design principles for retail Individual and micro, small and medium enterprise (MSME) customer segments Build innovative products and propositions backed with problem solving mindset to...

  • Machine learning Engineer

    3 weeks ago

    Gurugram, India Weekday AI Full time

    This role is for one of the Weekday's clientsMin Experience: 3 years Location: Gurugram, Delhi, Uttar Pradesh, NCR, NOIDA, Kanpur, Uttarakhand JobType: full-time We are seeking a highly skilled and motivated Machine Learning Engineer with 3–5 years of hands-on experience to join our growing team. The ideal candidate will have strong expertise in...

  • Machine learning Engineer

    1 week ago

    Gurugram, India Weekday AI Full time

    This role is for one of the Weekday's clientsMin Experience: 3 years Location: Gurugram, Delhi, Uttar Pradesh, NCR, NOIDA, Kanpur, Uttarakhand JobType: full-timeWe are seeking a highly skilled and motivated Machine Learning Engineer with 3–5 years of hands-on experience to join our growing team. The ideal candidate will have strong expertise in designing,...

  • Machine learning Engineer

    1 week ago

    Gurugram, India Weekday AI Full time

    This role is for one of the Weekday's clientsMin Experience: 3 years Location: Gurugram, Delhi, Uttar Pradesh, NCR, NOIDA, Kanpur, Uttarakhand JobType: full-timeWe are seeking a highly skilled and motivated Machine Learning Engineer with 3–5 years of hands-on experience to join our growing team. The ideal candidate will have strong expertise in...

  • Machine Learning Engineer

    1 week ago

    Gurugram, India GSPANN Full time

    Description GSPANN is hiring a Machine Learning/Data Engineer Specialist to build data pipelines, develop ML workflows, and deploy production models. Expertise in Python, Structured Query Language (SQL), and ML frameworks like TensorFlow, PyTorch, and Scikit-learn is required.Role and Responsibilities Design and implement scalable data pipelines and...

  • Machine Learning Engineer

    1 week ago

    Gurugram, India GSPANN Full time

    Description GSPANN is hiring a Machine Learning/Data Engineer Specialist to build data pipelines, develop ML workflows, and deploy production models. Expertise in Python, Structured Query Language (SQL), and ML frameworks like TensorFlow, PyTorch, and Scikit-learn is required.Role and Responsibilities Design and implement scalable data pipelines and...

  • Machine Learning Engineer

    3 weeks ago

    Gurugram, India PlusX Electric Full time

    We are seeking a highly skilledMachine Learning Developerto join our team. The ideal candidate will have strong expertise in developing, training, and deploying machine learning models usingPythonand frameworks such asTensorFlow, PyTorch, and Scikit-learn. The role requires the ability to work on end-to-end ML pipelines, from data preprocessing to model...

  • Machine Learning Engineer

    2 weeks ago

    Gurugram, Gurugram, India micro1 Full time

    Job Description Job Title: Machine Learning Engineer Job Type: Full-time About Us: Our mission at micro1 is to match the most talented people in the world with their dream jobs. If you are looking to be at the forefront of AI innovation and work with some of the fastest-growing companies in Silicon Valley, we invite you to apply for a role. By joining...

  • Machine Learning Engineer

    3 weeks ago

    Gurugram, India Accedepro Private Limited Full time

    We have an exciting job role for ML Engineer as below at Noida / Gurgaon. This is a WFO - Onsite role. Immediate Joiners Only. Prefer Locals Only. Machine Learning Engineer | 3-4 Years | Perm Role | Noida (or) Gurgaon || Roles & responsibilities The role involves developing Machine Learning models using Python for stock screening. Pre-built sanitized data...