AI Visual Specialist /  Machine Learning Engineer

AI Visual Specialist / Machine Learning Engineer
Remote with some visits to the London office / Hybrid also possible

The Role: Machine Learning Engineer (Visual Specialist)
We're looking for a hands-on Visual Specialist Machine Learning Engineer who lives and breathes generative AI—especially for image and video generation. You’ll work with bleeding-edge models (think SDXL, Flux, WAN 2.1), optimise inference for real-time performance, and help us scale our back-end for millions of users. You will be creating lifelike Ai Characters through text, voice and images. Working under the guidance of their AI Visual Lead, this role would be ideal for a someone with 3+ years of experience.

What You’ll Be Doing

• Optimise cutting-edge visual models (Flux, WAN 2.1, SDXL)
  
  
• Scale inference with low-latency, GPU-optimised infrastructure
  
  
• Train custom models on domain-specific datasets
  
  
• Write clean, scalable Python code for model and platform integration
  
  ⚠️ Note: You’ll be working with large multimedia datasets that include NSFW/adult content so must be happy working in this environment.
  
  Skills required:
  
  
• 3+ years in GenAI-focused ML engineering experience ideally on Multimodal AI (text, voice, image, video)
  
  
• A PhD or MSc from a top University in Machine learning or related subjects.
  
  
• Strong experience with visual AI models such as Stable Diffusion, Flux, WAN 2.1, or similar
  
  
• Fluency with PyTorch, JAX or TensorFlow
  
  
• GPU inference optimisation (CUDA, PEFT, Bits And Bytes)
  
  
• Docker/Kubernetes deployment experience desirable but not essential.
  
  
• Strong Python and back-end engineering skills
  
  
• Ability to deliver production-ready visual models, performance-constrained applications
  
  Location – Primarily remote (UK only) with some visits to the London office required / Hybrid working also available.
  
  Salary: £65,000 - £85,000 + benefits and Stock options possible
  
  Be part of a rapidly growing and well-funded AI startup, working on cutting edge, innovative visual technologies
  
  Apply today for immediate consideration and interview in the next week.
  
  N.B. You will need to be confortable working with Ai based NSFW / Adult content