AI Engineer

AI Engineer — Image‑to‑Video (Mid‑Level)

Location : Mumbai (on‑site / hybrid)

Contract : 6 months, extendable

Start : ASAP

What you'll do

• Build, fine‑tune, and ship image‑to‑video generation pipelines (prompt‑to‑video, storyboard‑to‑video, identity‑preserving headshots)
• Integrate and iterate on SOTA components (Stable Video Diffusion, AnimateDiff, LTX‑Video / 13B variants, CogVideo‑X, ControlNet‑style conditioning).
• Optimize inference for throughput and latency (TorchScript / ONNX, TensorRT, CUDA kernels, xFormers / Flash‑Attention, mixed precision).
• Handle multi‑GPU training / inference (DDP, gradient checkpointing, sharded weights, efficient sampling).
• Own dataset curation and augmentation for faces / motion; enforce consent, licensing, and privacy.
• Build evaluation loops and dashboards (FVD, CLIP / ID‑similarity, temporal consistency, face‑ID retention).
• Productionize with Docker and CI / CD; wire up tracking (W&B / ClearML) and experiment reproducibility.
• Collaborate with design and product to convert creative briefs into deployable features and A / B tests.

Must‑have

Nice‑to‑have

Tools you'll touch

What we offer

How to apply

Skills Required

Pytorch, Docker, Ffmpeg, Python, Cuda