Machine Learning Engineer (Diffusion/Vision) - Remote

Overview

We’re building AI systems for a global audience. This role is a global role with hybrid work arrangement - combining flexible remote work with in-office collaboration at our HQ.

In Short

• Train and customize diffusion models using LoRA, DreamBooth, and other methods.
• Build, clean, and annotate large-scale image datasets.
• Develop pipelines to measure fidelity, diversity, style adherence, and safety.
• Apply GPU memory optimization and distributed training for efficiency.
• Ship diffusion-powered features into production with monitoring.
• Collaborate with product and design to integrate generative vision capabilities.

Requirements

• Strong experience with diffusion models and generative vision.
• Hands-on skills with DreamBooth, LoRA/QLoRA, and fine-tuning methods.
• Proficiency with PyTorch.
• Experience in dataset preparation.
• Knowledge of GPU optimization and efficient training techniques.
• Strong foundations in software engineering and clean code practices.

Benefits

• Flat structure & real ownership.
• Full involvement in direction and consensus decision making.
• Flexibility in work arrangement.
• High-impact role with visibility across product, data, and engineering.
• Top-of-market compensation and performance-based bonuses.
• Global exposure to product development.
• Housing rental subsidies and quality cafeteria.
• Health, dental & vision insurance.
• Global travel insurance.
• Unlimited, flexible time off.