Back to jobs
Logo of Repli5
Repli5
Actively hiring

Machine Learning Engineer - Computer Vision

Workplace
On-site
Commitment
Full-time
Seniority
Mid-Level
Posted
Location
Gothenburg, Sweden

As a Machine Learning Engineer at Repli5, you will develop and fine-tune generative AI pipelines for synthetic image generation, design ML workflows for photorealistic datasets, collaborate with 3D artists and simulation engineers, work with customers in automotive, robotics, and aerospace, and optimize generation systems for scalability and speed.

Responsibilities

  • Develop and fine-tune generative AI pipelines (e.g., Stable Diffusion, ControlNet, LoRA) for synthetic image generation.
  • Design and implement machine learning workflows to generate structured, photorealistic datasets.
  • Collaborate with 3D artists and simulation engineers to align ML models with physical environments and sensor setups.
  • Work closely with customers in automotive, robotics, and aerospace to deliver tailored datasets.
  • Optimize generation systems for scalability, quality, and speed using tools like ComfyUI or custom pipelines.

Requirements

  • The Essentials: MSc or PhD in Computer Vision, Machine Learning, or related field with 3+ years of hands-on experience with deep learning frameworks (e.g., PyTorch, TensorFlow). Extensive experience with state-of-the-art image generation models (GAN, Diffusion, Transformers)
  • Exploring Mindset: Experience exploring workflows with generative AI models (especially diffusion-based) and a strong passion for reading and applying new research papers.
  • Practitioner: A track record of deploying ML systems in domains like industrial automation, defense, or automotive.
  • The Startup Gene: You excel in fast-paced, ambiguous environments where you can take initiative and work independently.
  • A Team Player: A problem-solving intellect with a team player spirit and a positive can-do attitude.

Nice to have

  • Toolset Curiosity (Big Plus!): Familiarity with tools like LoRA, ControlNet, ComfyUI, or custom workflows around Stable Diffusion.
  • Domain Knowledge (Big Plus!): Strong understanding of synthetic data generation and its role in training perception systems.

Benefits

  • A competitive compensation package.
  • Flexibility and work-life balance.
  • Challenging tasks that are at the forefront of technology.
  • The autonomy to make major technical decisions and influence the product direction.
  • An international, caring culture where it is natural to both ask for and offer help.
  • Flexible working hours and the possibility to work from home some days a week.
  • Fun at work.