Lexlegis.ai Logo

Lexlegis.ai

Generative AI Researcher – Image & Video Diffusion

Sorry, this job was removed at 06:12 a.m. (IST) on Thursday, Mar 19, 2026
Be an Early Applicant
In-Office
Mumbai, Maharashtra
In-Office
Mumbai, Maharashtra

Similar Jobs

4 Hours Ago
Hybrid
Mid level
Mid level
Big Data • Food • Hardware • Machine Learning • Retail • Automation • Manufacturing
Support and improve Data & Analytics products on Google Cloud, manage platform operations, implement SRE practices, and enhance observability and security.
Top Skills: BashCi/CdCloud MonitoringDatadogGCPGithub ActionsGrafanaJenkinsKubernetesPrometheusPythonSecurity ToolingTerraform
Junior
Big Data • Food • Hardware • Machine Learning • Retail • Automation • Manufacturing
Support data integrity and reliability by creating automated tests and managing data workflows on GCP. Troubleshoot and optimize SQL queries.
Top Skills: AutomicBigQueryCi/CdDataplexGCPPythonTricentis Tosca
4 Hours Ago
Hybrid
Senior level
Senior level
Big Data • Food • Hardware • Machine Learning • Retail • Automation • Manufacturing
The PLM Process Lead will optimize service delivery in SAP S/4 PLM, manage vendor relationships, lead projects, and ensure adherence to configuration standards.
Top Skills: It Infrastructure LibrarySap EccSap MdgSap S/4 Plm

We are looking for an Engineer to build the training infrastructure, data pipelines, and inference optimization systems for state-of-the-art Diffusion Transformer (DiT) models. This role focuses on scaling the fine-tuning and deployment of models like QwenWan, and LTX-2.

Key Responsibilities

  • Training Infrastructure: Design and maintain scalable pipelines for training and fine-tuning Diffusion Transformer models on large-scale GPU clusters.
  • Model Optimization: Optimize the inference performance of WanLTX-2, and Qwen (Vision) using quantization, pruning, and hardware-aware tuning (e.g., TensorRT, FlashAttention).
  • Data Engineering: Develop efficient ingestion and preprocessing pipelines for high-resolution image and video datasets used in generative tasks.
  • Capability Expansion: Implement engineering workflows that allow researchers to rapidly fine-tune and expand the capabilities of open-weights diffusion models.
  • Production Deployment: Transition experimental fine-tuned models into reliable, low-latency production services.
  • Resource Management: optimize distributed training jobs (FSDP, DeepSpeed) to maximize GPU utilization and minimize costs.

Required Qualifications

  • Min 2 years of experience in Machine Learning Engineering with a focus on generative models.
  • Core Tech: Strong proficiency in PyTorchJAX, and distributed training frameworks.
  • Model Expertise: Hands-on experience deploying or fine-tuning Diffusion Transformers (DiT) and specifically Qwen (Image)Wan, or LTX-2.
  • Architecture: Deep understanding of Transformer-based diffusion backbones and flow matching (removing legacy reliance on CNNs/RNNs).
  • Tooling: Proficiency in Python and modern ML ecosystem tools (e.g., Hugging Face, Diffusers, FFmpeg for video processing).
  • Compute: Experience debugging and optimizing workloads in multi-node GPU environments.

Preferred Qualifications

  • Inference Optimization: Experience with techniques like KV-caching, compile-time optimizations, or kernel fusion for transformers.
  • MLOps: Familiarity with experiment tracking (W&B) and model versioning tools in a generative media context.
  • Streaming: Experience handling real-time video generation or streaming inference pipelines.
  • Open Source: Contributions to libraries like diffusers or active experimentation with the latest open-source DiT implementations.

What you need to know about the Pune Tech Scene

Once a far-out concept, AI is now a tangible force reshaping industries and economies worldwide. While its adoption will automate some roles, AI has created more jobs than it has displaced, with an expected 97 million new roles to be created in the coming years. This is especially true in cities like Pune, which is emerging as a hub for companies eager to leverage this technology to develop solutions that simplify and improve lives in sectors such as education, healthcare, finance, e-commerce and more.

Sign up now Access later

Create Free Account

Please log in or sign up to report this job.

Create Free Account