Machine Learning Engineer (Diffusion/Vision) - Remote

Overview

Transform Visual Models into Real-World Applications. We’re building AI systems for a global audience. We are living in an era of AI transition - this new project team will be focusing on building applications to enable more real world impact and highest usage for the world.

In Short

• Fine-tune & Adapt – Train and customize diffusion models (SDXL, Flux, Stable Diffusion variants) using LoRA, DreamBooth, and other parameter-efficient methods.
• Curate Datasets – Build, clean, and annotate large-scale image datasets with captioning, tagging, and NSFW filtering for safe and aligned generation.
• Evaluate & Align – Develop pipelines to measure fidelity, diversity, style adherence, and safety across generated outputs.
• Optimize Performance – Apply GPU memory optimization, latent diffusion tricks, and distributed training for efficient scaling.
• Deploy & Monitor – Ship diffusion-powered features into production with monitoring for drift, latency, and quality.
• Collaborate & Deliver – Work with product and design to integrate generative vision capabilities into user experiences.

Requirements

• Strong experience with diffusion models and generative vision (Stable Diffusion, SDXL, Flux, etc.).
• Hands-on skills with DreamBooth, LoRA/QLoRA, and fine-tuning methods.
• Proficiency with PyTorch (preferred).
• Experience in dataset preparation (captioning, tagging, filtering, augmentation).
• Knowledge of GPU optimization, latent diffusion, and efficient training techniques.
• Strong foundations in software engineering, algorithms, 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
• Lots of perks - housing rental subsidies, a quality company cafeteria, and overtime meals
• Health, dental & vision insurance
• Global travel insurance (for you & your dependents)
• Unlimited, flexible time off