Machine Learning Engineer (Diffusion / Vision) - Remote

Overview

This role focuses on transforming visual models into real-world applications by building AI systems for a global audience. You'll work closely with various teams to build and scale impactful AI solutions.

In Short

• Fine-tune and adapt diffusion models using parameter-efficient methods.
• Curate large-scale image datasets for safe and aligned generation.
• Develop pipelines to measure fidelity and safety across outputs.
• Apply GPU memory optimization for efficient scaling.
• Deploy diffusion-powered features into production with monitoring.
• Collaborate with product and design teams to integrate capabilities.

Requirements

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

Benefits

• Flat structure with real ownership.
• Flexibility in work arrangement.
• High-impact role with visibility across teams.
• Top-of-market compensation and performance-based bonuses.
• Global exposure to product development.
• Health, dental, and vision insurance.
• Unlimited, flexible time off.