Description

You focus on training advanced models for image and video processing, analyzing design choices, and evaluating the performance of neural network architectures.

Requirements

• •Training large scale Diffusion models
• •Finetuning Diffusion models for applications
• •Deep understanding of evaluating generative models
• •Strong proficiency in PyTorch and NN architectures
• •Deep understanding of training techniques
• •Experience with writing Triton kernels
• •Profiling and optimizing GPU operations

Tasks

• •Train large-scale diffusion models for images and videos.
• •Ablate design choices and communicate results with the team.
• •Evaluate speed and quality trade-offs of neural network architectures.

Tools & Technologies