Research Engineer

What you'll do

• Implement generative AI features and build image and video data pipelines for model training.
• Improve model quality through user feedback and preference-based tuning, including try-on for diverse body types and styles.
• Develop personalized video and 3D try-on capabilities and collaborate in a shared codebase.

What they ask for

• Hands-on experience training and fine-tuning diffusion models (must-have).
• Familiarity with implementing and optimizing diffusion model training pipelines.
• Experience implementing ML papers from scratch in pytorch or jax.
• Ability to work independently and effectively in a rigorous, fast-paced environment.

Nice to have

• Experience with virtual try on.
• Experience building consumer products.

Tools

diffusion models, pytorch, jax

AI focus

Build and improve diffusion-based generative AI models for avatars and personalized virtual try-on, using image and video datasets and user feedback.