Remote Machine Learning Engineer (Diffusion / Vision)

at Bjak

Posted 15 hours ago 1 applied

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.

Get realtime job alerts

Be the first to know about new jobs