Machine Learning Engineer (Diffusion/Vision) - Remote

Overview

This role involves transforming visual models into real-world applications, focusing on developing AI systems that have significant impact and high usage.

In Short

• Fine-tune diffusion models using LoRA, DreamBooth, and other methods.
• Build and annotate large-scale image datasets.
• Develop evaluation frameworks for model outputs.
• Optimize GPU memory and training techniques.
• Deploy generative vision features into production.
• Collaborate with product and design teams.

Requirements

• Experience with diffusion models (Stable Diffusion, SDXL, Flux).
• Hands-on skills with DreamBooth and LoRA/QLoRA.
• Proficiency with PyTorch.
• Experience in dataset preparation and augmentation.
• Knowledge of GPU optimization techniques.
• Strong software engineering foundations.

Benefits

• Flat organizational structure with real ownership.
• Involvement in product direction and decision making.
• Flexible hybrid work arrangement.
• High-impact role with visibility across teams.
• Competitive compensation and performance bonuses.
• Global exposure to product development.
• Various perks including housing subsidies and health insurance.
• Unlimited, flexible time off.