Senior ML Scientist (GenAI)

TL;DR

Senior ML Scientist (GenAI): Invent and deploy advanced generative AI models and features for image and video creation with an accent on diffusion, transformer, and multimodal text-to-image/video models. Focus on designing scalable training and inference pipelines, collaborating across teams, and publishing research with real-world impact.

What you will do

• Invent and publish advanced diffusion, transformer, and multimodal text-to-image/video models including high-resolution generation up to 16K.
• Build innovative features for image/video retouching, effects, quality enhancement, and avatar generation.
• Develop and optimize training and inference pipelines using data parallelism, quantization, distillation, and TensorRT.
• Design, build, and maintain training and evaluation workflows with PyTorch 2.x, DataBricks, and SLURM.
• Create human-in-the-loop evaluations including visual comparisons and technical metrics.
• Collaborate closely with product, design, and engineering teams to ship research-driven features at scale.

Requirements

• Ph.D. or equivalent research experience in Computer Science, Electrical Engineering, Mathematics, or related field.
• 5+ years experience delivering computer vision and generative AI models in production.
• First-author publications or patents in top-tier venues such as CVPR, ICCV, ECCV, SIGGRAPH, NeurIPS, or ICLR.
• Deep expertise in PyTorch and CUDA.
• Hands-on experience with diffusion models, GANs, and vision-language architectures.
• Strong foundations in linear algebra, probability, optimization, and large-scale model training techniques.

Nice to have

• Contributions to open-source ML/CV projects or toolkits.