ML Engineer – Virtual Try On + Computer Vision

5 years

0 Lacs

Posted:2 days ago| Platform: Linkedin logo

Apply

Work Mode

Remote

Job Type

Full Time

Job Description

Location:

Type:

Company:


About the Role

India’s most advanced virtual try-on engine


ML Engineer (Computer Vision + Try-On)


You’ll have full autonomy to build, experiment, and ship — working directly with React, Spring Boot, DevOps, and design folks already in place.


This is not a junior researcher role. This is 

one person building the brain of the system

 - and setting the foundation for India's biggest visual shopping innovation.What You’ll Build

Stage 1: User Image Preprocessing

  • Human parsing (face, body, hair), pose detection, face/limb alignment
  • Auto orientation, canvas resizing, brightness/contrast normalization

Stage 2: Product Image Processing

  • Background removal, garment segmentation (SAM/U^2-Net/YOLOv8)
  • Handle occlusions, transparent clothes, long sleeves, etc.

Stage 3: Try-On Engine

  • Implement and iterate on CP-VTON / TryOnDiffusion / FlowNet
  • Fine-tune on custom data for realism, garment drape, identity retention

Inference Optimisation

  • TorchScript / ONNX, batching, inference latency minimization
  • Collaborate with DevOps for Lambda/EC2 + GPU deployment

Postprocessing

  • Alpha blending, edge smoothing, fake shadows, cloth-body warps


You’re a Fit If You:
  • Have 

    2–5 years

     in ML/CV with real shipped work (not just notebooks)
  • Have worked on: 

    human parsing, pose estimation, cloth warping, GANs

  • Are 

    hands-on with PyTorch

    , OpenCV, Segmentation Models, Flow or ViT
  • Can replicate models from arXiv fast, and care about output quality
  • Want to 

    own a system seen by millions

    , not just improve metrics


Stack You’ll Use
  • PyTorch, ONNX, TorchScript, Hugging Face
  • DensePose, OpenPose, Segment Anything, Diffusion Models
  • Docker, Redis, AWS Lambda, S3 (infra is already set up)
  • MLflow or DVC (can be implemented from scratch)


For exceptional talent, we’re flexible on cash vs equity split.
Why This Is a Rare Opportunity
  • Build the core AI product

     that powers a breakout consumer app
  • Work in a 

    zero BS, full-speed team

     (React, SpringBoot, DevOps, Design all in place)
  • Be 

    the founding ML brain

     and shape all future hires
  • Ship in weeks, not quarters — and see your output in front of users instantly


Apply now, or DM Dheekshith (Founder) on LinkedIn with your GitHub or project links.

Let’s build something India’s never seen before.

Mock Interview

Practice Video Interview with JobPe AI

Start DevOps Interview
cta

Start Your Job Search Today

Browse through a variety of job opportunities tailored to your skills and preferences. Filter by location, experience, salary, and more to find your perfect fit.

Job Application AI Bot

Job Application AI Bot

Apply to 20+ Portals in one click

Download Now

Download the Mobile App

Instantly access job listings, apply easily, and track applications.

coding practice

Enhance Your Skills

Practice coding challenges to boost your skills

Start Practicing Now

RecommendedJobs for You