Remote Machine Learning Engineer (Diffusion / Vision)

Description:

• The role involves transforming visual models into real-world applications, focusing on building AI systems for a global audience.
• This position is remote and requires collaboration with regional teams across product, engineering, operations, infrastructure, and data to build and scale impactful AI solutions.
• Responsibilities include fine-tuning state-of-the-art models, designing evaluation frameworks, and bringing AI features into production to ensure models are intelligent, safe, trustworthy, and impactful at scale.
• Key tasks include training and customizing diffusion models using methods like LoRA and DreamBooth, curating large-scale image datasets, developing evaluation pipelines, optimizing performance, deploying features into production, and collaborating with product and design teams.

Requirements:

• Candidates must have strong experience with diffusion models and generative vision, including technologies like Stable Diffusion, SDXL, and Flux.
• Hands-on skills with DreamBooth, LoRA/QLoRA, and fine-tuning methods are required.
• Proficiency in PyTorch is preferred.
• Experience in dataset preparation, including captioning, tagging, filtering, and augmentation, is necessary.
• Knowledge of GPU optimization, latent diffusion, and efficient training techniques is essential.
• A strong foundation in software engineering, algorithms, and clean code practices is required.

Benefits:

• The position offers a flat structure with real ownership and full involvement in direction and consensus decision-making.
• There is flexibility in work arrangements and a high-impact role with visibility across product, data, and engineering.
• Compensation is top-of-market with performance-based bonuses.
• Employees receive global exposure to product development and various perks, including housing rental subsidies, a quality company cafeteria, and overtime meals.
• Health, dental, and vision insurance are provided, along with global travel insurance for employees and their dependents.
• The company offers unlimited, flexible time off.