About this role
Generative AI Researcher – Image & Video Diffusion at Lexlegis AI. Location: Mumbai, Maharashtra, India. Role: training models, optimizing inference, developing pipelines Requirements: 2+ years ML engineering experience with generative models; strong PyTorch/JAX skills; distributed training and multi-node GPU optimization; experience fine-tuning/deploying DiT/Qwen/Wan/LTX-2; proficiency with Python and modern ML tooling. Category: Research and Development (R&D) Seniority: Entry Level Tools: PyTorch, JAX, TensorRT, FlashAttention, FSDP, DeepSpeed, Python, Hugging Face, Diffusers, FFmpeg, W&B, Diffusion Transformer, Qwen, Wan, LTX-2 Commitment: Full Time Workplace: Onsite Languages: English