Machine Learning Engineer (Diffusion/Vision) - Remote

Overview

You will be transforming visual models into real-world applications by building AI systems for a global audience.

In Short

• Fine-tune and adapt diffusion models using efficient methods.
• Build, clean, and annotate large-scale image datasets.
• Develop evaluation pipelines to measure output quality.
• Optimize performance with GPU memory and distributed training.
• Deploy features into production and monitor their performance.
• Collaborate with product and design teams to enhance user experience.

Requirements

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

Benefits

• Flat organizational structure with real ownership.
• Full involvement in decision-making.
• Flexible work arrangements.
• 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 and flexible time off.