Tauhid Khan

ML Research Engineer · Mumbai

Generative media
and robot learning.

I turn ML research into production image and video systems at Fynd. My independent work explores how robots learn to perceive and act.

Tauhid Khan
TAUHID KHANMUMBAI, IN ↗
Noise → image

GENERATIVE MEDIA

Recorded sampling stages from my flow-matching image experiment.

t = 0.00
0 · NOISE1 · OUTPUT
Explore image generation ↗

Image: recorded sampling stages. Action chunks & particle flow: illustrative.

AT THE INTERSECTION OFGenerative AIComputer visionRobot learning

Experience

Where I’ve worked

Generative media at Fynd.
Computer vision at Wobot.

SEP 2023 — PRESENT

Fynd / ShopSense

ML Research Engineer
Illustration of an image forming from noise, a pixelated image becoming sharper, and subject masks across video frames.
Image generation, super-resolution & video segmentation · illustration

Building production generative image and video systems, from distributed model training to optimized GPU inference.

  • Multi-GPU fine-tuning of SDXL and Flux.Dev for controllable generation, using representation alignment, distillation, and LoRA for efficient adaptation and deployment.
  • Flow-matching super-resolution, including distillation to a single-step generator.
  • Video segmentation, inpainting, and restoration pipelines up to 4K.
~20Kdaily active users
across AI media services
Up to 56%lower latency in
foreground removal
More engineering detail

Curated multimodal training data with VLM-assisted captioning and filtering. Optimized tiled, 4× super-resolution inference on NVIDIA L4; TensorRT conversion reduced latency by approximately 40–45% in that pipeline.

FEB 2022 — SEP 2023

Wobot Intelligence

Computer Vision Engineer I
Illustration of a retail store and drive-through, with detection boxes around a shopper and queued cars.
Retail & drive-through vision · illustration

Built multi-camera person and vehicle tracking for drive-through journey analytics and production video inference.

  • Detection and tracking pipelines using YOLO, DeepSORT, and ByteTrack.
  • Reusable deployment workflows with Docker and NVIDIA Triton.
  • Reworked multithreaded video processing, reducing CPU utilization from 90% to 40%.
1,000+cameras supported
250+deployment locations

Selected work

Projects and experiments

Open research, practical experiments,
and the code behind them.

07 PROJECTS
NOISE → IMAGESTEP 7 / 7
Four generated face samples from the flow-matching experiment, progressing from noise to images
Actual samples from the repository.

02 / GENERATIVE MODELING

Flow-based Models

Exploring how noise becomes an image. Conditional flow matching with optimal transport paths and flexible ODE sampling.

OT-CFMPyTorchEuler / Heun
Explore the code
Dexterity Lab interface showing a simulated robotic hand, grasping scene, and control panelsBROWSER-BASED SIMULATION

03 / ROBOTICS & INTERACTION

Dexterity Lab

A robotic hand playground in the browser. Camera-based hand tracking, physics-driven grasping, and inverse-kinematics pinch control.

Three.jsRapierMediaPipe
Explore the code
LOW LIGHT → ENHANCEDZERO-DCE LITE
Zero-DCE Lite enhanced image revealing two chairs and curtains in a room Original low-light image of the same room, with the chairs and curtains in shadow Original Enhanced
Original and enhanced output from the repository.

04 / IMAGE ENHANCEMENT

Zero-DCE

TensorFlow implementations of Zero-DCE and Zero-DCE++ for low-light image enhancement, with TensorFlow Lite conversion.

TensorFlowZero-reference learningTFLite
Explore the code

06 / TINYML & AUDIO

Audio Classifier

Final-year degree project · B.Sc. Computer Science · 2022

TinyML sound classification using a quantized CNN and TensorFlow Lite Micro on Arduino Nano 33 BLE Sense.

INT8 quantizationTFLite MicroArduino
Explore the code

07 / ROBOT CONTROL

UR5e Pose Control

From a target pose to a planned motion. Python control with ROS 2 and MoveIt 2, demonstrated with mock hardware and MuJoCo physics.

ROS 2MoveIt 2RVizMuJoCo
Explore the code
BUILT TO BE EXPLORED.More on GitHub

Research

A straighter path
from noise to data.

ICML 2026 · SPIGM WORKSHOP2026

Isokinetic Flow Matching for Pathwise Straightening of Generative Flows

Tauhid Khan

A lightweight, Jacobian-free regularizer that encourages straighter generative trajectories for more accurate few-step sampling.

What’s the idea?

Even when conditional flow-matching paths are straight, the learned marginal velocity field can curve. The method penalizes pathwise acceleration through a self-guided finite-difference estimate, without auxiliary encoders or second-order autodifferentiation.

Accepted at the Workshop on Structured Probabilistic Inference & Generative Modeling at ICML 2026.

A little about me

How I like to work

My independent work means getting hands-on with the whole loop—assembling a robot, collecting demonstrations, training policies, and understanding where they fail.

EDUCATION
B.Sc. Computer ScienceUniversity of Mumbai · 2022 · CGPA 9.21 / 10
TOOLS I WORK WITH

PyTorch / TensorFlow / OpenCV / TensorRT / Triton / ManiSkill / MuJoCo / LeRobot / Docker

Résumé

My experience across AI research, engineering, and robot learning.

LET’S CONNECTRESEARCH ↔ ENGINEERING

Let’s talk.

Working on generative AI or robot learning?
I’d love to hear what you’re building.