Enable job alerts via email!

Machine Learning Engineer (Visual Specialist)

OhChat

Greater London

Remote

GBP 60,000 - 90,000

Full time

Yesterday

Be an early applicant

Boost your interview chances

Create a job specific, tailored resume for higher success rate.

Job summary

A leading company in the AI industry is seeking a Machine Learning Engineer to develop and optimise AI models for visual content generation. The role involves working with deep learning technologies and requires expertise in Python and machine learning frameworks. This position offers the chance to shape AI strategies in a remote-first environment, focusing on innovative solutions and user engagement.

Benefits

Competitive salary

Remote-first work environment

Ownership in AI strategy

Qualifications

3+ years of experience in machine learning engineering with a focus on GenAI.
Strong expertise in Stable Diffusion and Flux architectures.

Responsibilities

Fine-tune and optimise generative models.
Deploy and optimise models for low latency on GPU infrastructure.
Write clean, scalable Python code for back-end improvements.

Skills

Python

Machine Learning

Deep Learning

GPU Inference Optimisation

Education

Bachelor's Degree

Tools

PyTorch

Docker

Kubernetes

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.

About Oh

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.

The Role

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.

What You’ll Do

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.

What We’re Looking For

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

Why Join Us?

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.

Seniority level

Seniority level
Mid-Senior level

Employment type

Employment type
Full-time

Job function

Job function
Engineering and Information Technology
Industries
Technology, Information and Internet

Referrals increase your chances of interviewing at OhChat by 2x

Get notified about new Machine Learning Engineer jobs in London Area, United Kingdom.

London, England, United Kingdom 8 hours ago

London, England, United Kingdom 2 months ago

Machine Learning Scientists and Engineers: AI for Quantum

London, England, United Kingdom 2 weeks ago

Research Engineer, ML, AI & Computer Vision