Talent.com
This job offer is not available in your country.
Machine Learning Engineer

Machine Learning Engineer

Shiprocketgurugram, India
3 hours ago
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 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.
  • Create a job alert for this search

    Machine Learning Engineer • gurugram, India

    Related jobs
    • Promoted
    Senior Generative AI Engineer (Databricks,Data Lake)

    Senior Generative AI Engineer (Databricks,Data Lake)

    AmpstekDelhi, IN
    Title : Senior Generative AI Engineer (Databricks,Data Lake).Design and implement GenAI models (LLMs, multimodal, embeddings, and fine-tuning) for enterprise use cases. Architect and optimize data pi...Show moreLast updated: 1 day ago
    • Promoted
    Founding Sr. Machine Learning Engineer : No Salary Only Equity

    Founding Sr. Machine Learning Engineer : No Salary Only Equity

    Truxt.aiGhaziabad, IN
    As an early-stage company driven by a meaningful mission, we seek exceptional engineering talent who values purpose, long-term impact, and significant equity ownership over traditional salary compe...Show moreLast updated: 21 days ago
    • Promoted
    Machine Learning Engineer

    Machine Learning Engineer

    Branch InternationalDelhi, IN
    Branch delivers world-class financial services to the mobile generation.With offices in the United States, Nigeria, Kenya, and India, Branch is a for-profit socially conscious company that uses the...Show moreLast updated: 30+ days ago
    • Promoted
    AI / ML Engineer

    AI / ML Engineer

    VeltrisDelhi, IN
    Veltris is a Digital Product Engineering Services partner committed to driving technology-enabled transformation across enterprises, businesses, and industries. We specialize in delivering next-gene...Show moreLast updated: 19 days ago
    • Promoted
    • New!
    Machine Learning Engineer

    Machine Learning Engineer

    Kitronoida, India
    In this role, you’ll focus on increasing model accuracy, retraining and maintaining the ML pipeline, and supporting code and infrastructure upkeep. You’ll also drive R&D of innovative products like ...Show moreLast updated: 3 hours ago
    • Promoted
    Machine Learning Engineer

    Machine Learning Engineer

    DexianGhaziabad, IN
    We seek a Machine Learning Engineer colleague to join our world-class client, one of the.Employment / Contract Type : Contract. Duration : 1 year (with potential to be extended).Hours per Week : 40 (Full...Show moreLast updated: 27 days ago
    • Promoted
    AI / Machine Learning Engineer (India) – Mid to Senior

    AI / Machine Learning Engineer (India) – Mid to Senior

    Epsilon Technologies GroupDelhi, IN
    Epsilon Technologies Group (Epsilon) is a specialist capital markets technology solutions and services firm helping financial institutions modernize their platforms, risk management, and operations...Show moreLast updated: 19 days ago
    • Promoted
    AI LLM Research Engineer (TTS, M-LLM, Agentic Workflow)

    AI LLM Research Engineer (TTS, M-LLM, Agentic Workflow)

    FlashIntelDelhi, IN
    FlashIntel is seeking a dedicated and innovative Research Engineer with a focus on Multimodal Large Language Models (m-LLMs), Text-to-Speech (TTS) technologies, and agentic workflows.This position ...Show moreLast updated: 30+ days ago
    • Promoted
    Senior Machine Learning Engineer

    Senior Machine Learning Engineer

    Elife TransferDelhi, IN
    A fast-growing start-up headquartered in San Francisco, CA, USA in the heart of Silicon Valley.We recruit worldwide as our customer base is global. Reliable ground transportation provider, any type ...Show moreLast updated: 26 days ago
    • Promoted
    Machine Learning Engineer (Relocation to Munich possible)

    Machine Learning Engineer (Relocation to Munich possible)

    DRIMCO GmbHGhaziabad, IN
    At DRIMCO, we are driving the future of industrial automation and engineering by developing intelligent AI agents tailored for the manufacturing and automotive sectors. As part of our growing team, ...Show moreLast updated: 30+ days ago
    • Promoted
    Manager of Machine Learning

    Manager of Machine Learning

    FirstsourceDelhi, IN
    Required Experience : 7+ Years.Firstsource Solutions Limited, an RP-Sanjiv Goenka Group company (NSE : FSL, BSE : 532809, Reuters : FISO. BO, Bloomberg : FSOL : IN), is a specialized global business proce...Show moreLast updated: 8 days ago
    • Promoted
    AIML Engineer

    AIML Engineer

    IntraEdgeDelhi, IN
    Artificial Intelligence (AI / ML).AI-powered enterprise applications.The engineer will be responsible for developing scalable. Design and develop enterprise-grade applications in.NLP, predictive analy...Show moreLast updated: 1 day ago
    • Promoted
    Data / ML Engineer

    Data / ML Engineer

    GleantapDelhi, IN
    Gleantap is a customer engagement platform powering fitness, wellness, and service businesses.Python (scikit-learn, LightGBM, XGBoost). Set up retraining and monitoring pipelines (Airflow, MLflow, o...Show moreLast updated: 23 days ago
    • Promoted
    Machine Learning Engineer

    Machine Learning Engineer

    Gabeo.aiDelhi, India, India
    Please apply only if you're available to start right away.If you're currently serving a notice period or cannot join immediately, we kindly ask that you refrain from applying.We're only interested ...Show moreLast updated: 30+ days ago
    • Promoted
    • New!
    Machine Learning Engineer

    Machine Learning Engineer

    Clarivatenoida, India
    We are looking for a ML Ops Engineer to join our Technology team at Clarivate.You will get the opportunity to work in a cross-cultural work environment while working on the latest web technologies ...Show moreLast updated: 3 hours ago
    • Promoted
    Lead Artificial Intelligence Engineer

    Lead Artificial Intelligence Engineer

    TalentXM (Formerly BlockTXM Inc)Delhi, IN
    We are looking for a visionary and technically skilled.Lead AI Engineer – Machine Learning.In this leadership role, you will guide a team of AI / ML engineers, contribute hands-on to critical.This ro...Show moreLast updated: 19 days ago
    • Promoted
    • New!
    Machine Learning Engineer

    Machine Learning Engineer

    GSPANNgurugram, India
    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 (...Show moreLast updated: 3 hours ago
    • Promoted
    AI / ML Engineer

    AI / ML Engineer

    Optimal Virtual EmployeeDelhi, IN
    Here is the Job Description as discussed : .As the AI / ML Engineer, you will be a key architect in the development of our . You will work alongside the founding team to integrate .Your contributions wi...Show moreLast updated: 1 day ago
    • Promoted
    Generative AI Engineer (LLM Expert)

    Generative AI Engineer (LLM Expert)

    BigRioDelhi, IN
    Generative AI Engineer (LLM Expert).BigRio is a Boston-based, remote-first technology consulting firm specializing in advanced data and software solutions. We partner with forward-thinking organizat...Show moreLast updated: 30+ days ago
    • Promoted
    • New!
    Machine Learning Engineer

    Machine Learning Engineer

    White Forcenoida, India
    We are seeking an experienced Machine Learning to join our AI-driven.The ideal candidate will have a strong background in prompt. Tree of Thought (ToT) and Chain of.Thought (CoT), along with hands-o...Show moreLast updated: 3 hours ago