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.