Machine Learning Engineer (Visual Specialist) (London Area)

Oh is on a mission to become the OpenAI of the spicy content industry, building a full-spectrum ecosystem of revolutionary AI infrastructure and products. Our platform, OhChat, features digital twins of real-world personalities and original AI characters, enabling users to interact with lifelike AI-generated characters through text, voice, and images, with a roadmap that includes agentic superModels, API integrations, and video capabilities.

We’re looking for a Machine Learning Engineer with deep expertise to own, optimise, and scale our AI models, ensuring cutting-edge performance and user engagement.

We are seeking a Machine Learning Engineer with a focus on back-end Python engineering visual asset generation to join our team. The ideal candidate will have hands-on experience in working with deep learning models for image and video generation, leveraging technologies such as SDXL, Flux, and WAN 2.1. You should be familiar with popular machine learning frameworks like PyTorch, Diffusers, Transformers, and Docker. As a member of our team, you'll work closely with researchers, engineers, and creatives to develop and deploy solutions that generate high-quality visual content, as well as playing a key part in building the back-end of our platform.

✅ Model Development & Optimisation: Fine-tune and optimise open-source and proprietary generative models (e.g., Flux, Wan 2.1).

✅ Inference Optimisation & Scaling: Deploy and optimise models optimised for low latency and cost-effective inference on GPU-based infrastructure .

✅ Custom Model Training: Train models with domain-specific datasets, improve response quality, reduce hallucinations, and enhance alignment with user intent.

✅ Back-end engineering: Write clean, scalable Python code to expand and improve our back-end

✅ Security & Compliance: Ensure AI models meet GDPR and ethical AI guidelines for content generation.

*Important note: As a part of this role you will be responsible for managing and maintaining large multimedia datasets of NSFW/adult content.

🚀 3+ years of experience in machine learning engineering with a focus on GenAI

🚀 Strong expertise in Stable Diffusion, Flux, and similar architectures

🚀 Experience fine-tuning and optimising visual models

🚀 Hands-on experience with PyTorch, JAX, or TensorFlow

🚀 Proficiency in GPU inference optimisation (CUDA, BitsAndBytes, PEFT)

🚀 Experience deploying ML models using Kubernetes and/or Docker

🚀 Strong experience with Python and software engineering best practices

🚀 Experience integrating visual models into real-world applications with performance constraints

Bonus if you have:

➕ Experience working in multimodal AI (text, voice, image, or video models)

➕ Experience working as a full-stack engineer (especially back-end and DevOps-focused)

➕ Background in AI-driven gaming, adult content, or interactive digital experiences

➕ Interest in crypto, Web3, or NFT-based AI models

🌍 Remote-first

💰 Competitive salary

🚀 Work on cutting-edge models powering next-gen AI interactions

🎯 Ownership & autonomy in shaping our AI strategy

🛠 Hands-on role in a company that ships fast and iterates rapidly

💡 Experiment with state-of-the-art AI models & inference techniques

This is a unique opportunity to lead the AI development of one of the most exciting AI companies in the companionship space. If you're passionate about building, optimising, and scaling AI-driven characters, we’d love to hear from you.