Specialist Machine Learning Engineer

JR United Kingdom; London; Remote; GBP 60,000 - 90,000; Full time

Job summary

A leading company in the AI industry is seeking a Specialist Machine Learning Engineer to optimize and scale AI models for their innovative platform. This role involves working with cutting-edge technologies to develop high-quality visual content and ensuring compliance with ethical guidelines. Join a remote-first team that values ownership and autonomy in shaping AI strategies.

Benefits

Remote-first

Competitive salary

Ownership & autonomy in shaping AI strategy

Hands-on role in a fast-paced company

Qualifications

• 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 optimizing visual models.

Responsibilities

• Fine-tune and optimize open-source and proprietary generative models.
• Deploy and optimize models for low latency and cost-effective inference.
• Write clean, scalable Python code to improve back-end.

Skills

Machine Learning Engineering

Python

GPU Inference Optimization

Model Optimization

Deep Learning

Tools

PyTorch

Docker

Kubernetes

Job description

Social network you want to login/join with:

Specialist Machine Learning Engineer, London

Client: OhChat

Location: London, United Kingdom

Job Category: Other

EU work permit required: Yes

Job Views: 2

Posted: 21.05.2025

Expiry Date: 05.07.2025

Job Description:

About OhChat 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.

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.

Responsibilities

• Model Development & Optimization: Fine-tune and optimise open-source and proprietary generative models (e.g., Flux, Wan 2.1).
• Inference Optimisation & Scaling: Deploy and optimise models 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.