Machine Learning Engineer (Visual Specialist)

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

Job summary

A leading company in AI-driven digital personalities is seeking a Machine Learning Engineer to develop and optimize deep learning models for image and video generation. This role emphasizes back-end Python engineering and compliance with GDPR. You'll work in a remote-first environment with cutting-edge AI technologies, ensuring ownership and autonomy in AI strategy.

Benefits

Remote-first work environment

Competitive salary

Opportunity to work with cutting-edge AI models

Ownership and autonomy in AI strategy

Fast-paced development and iteration

Qualifications

• At least 3 years of experience in machine learning engineering focused on GenAI.
• Expertise in Stable Diffusion, Flux, and similar architectures.

Responsibilities

• Model development and optimization of generative models.
• Deploying and optimizing models for low latency inference on GPU infrastructure.
• Writing scalable, clean Python code for back-end development.

Skills

Machine Learning

Python

GPU Inference Optimization

Tools

PyTorch

TensorFlow

Kubernetes

Docker

Job description

Job Description:

We are seeking a Machine Learning Engineer with expertise in visual asset generation to join our team at OhChat, a platform creating AI-driven digital personalities. The role involves developing, optimizing, and deploying deep learning models for image and video generation, with a focus on back-end Python engineering.

Responsibilities include:

1. Model development and optimization of generative models like Flux and Wan 2.1.
2. Deploying and optimizing models for low latency inference on GPU infrastructure.
3. Training models with domain-specific datasets to improve quality and alignment.
4. Writing scalable, clean Python code for back-end development.
5. Ensuring AI models comply with GDPR and ethical guidelines, managing large multimedia datasets.

Qualifications:

1. At least 3 years of experience in machine learning engineering focused on GenAI.
2. Expertise in Stable Diffusion, Flux, and similar architectures.
3. Experience with PyTorch, JAX, or TensorFlow.
4. Proficiency in GPU inference optimization techniques like CUDA.
5. Experience with deploying models via Kubernetes or Docker.
6. Strong Python skills and software engineering practices.

Preferred:

• Experience with multimodal AI (text, voice, image, video).
• Background in AI-driven gaming, adult content, or interactive digital experiences.
• Interest in crypto, Web3, or NFT-based AI models.

Benefits:

• Remote-first work environment.
• Competitive salary.
• Opportunity to work with cutting-edge AI models.
• Ownership and autonomy in AI strategy.
• Fast-paced development and iteration.

This role offers a chance to lead AI development in a pioneering company in the AI companionship space, focusing on building and scaling AI-driven characters.