Software Engineer - AI Systems & Infrastructure

ZipRecruiter; London; On-site; GBP 60,000 - 80,000; Full time

Job summary

A leading recruitment platform in London is seeking a software engineer to build and optimize infrastructure for large-scale AI model training and deployment. Responsibilities include developing training pipelines, optimizing model performance, collaborating with ML teams, and building deployment tools. Ideal candidates have strong programming skills and experience with ML frameworks. Competitive compensation and growth opportunities offered.

Responsibilities

• Develop scalable, distributed training and evaluation pipelines.
• Optimize model performance and system efficiency across different compute environments.
• Collaborate with ML teams to integrate models into production systems.
• Build tooling for deployment, monitoring, and continuous improvement of AI models.

Skills

Strong programming skills (Python, C++, or similar)

Experience with ML frameworks (e.g. PyTorch, TensorFlow, JAX)

Familiarity with cloud infrastructure and distributed systems

Solid understanding of performance optimization and system design

Job description

Overview

We're looking for a software engineer to help build and optimize the infrastructure behind large-scale AI model training and deployment. This role involves working across the full ML lifecycle - from data processing and training pipelines to efficient inference in production environments.

What You'll Do

• Develop scalable, distributed training and evaluation pipelines
• Optimize model performance and system efficiency across different compute environments
• Collaborate with ML teams to integrate models into production systems
• Build tooling for deployment, monitoring, and continuous improvement of AI models

What We're Looking For

• Strong programming skills (Python, C++, or similar)
• Experience with ML frameworks (e.g. PyTorch, TensorFlow, JAX)
• Familiarity with cloud infrastructure and distributed systems
• Solid understanding of performance optimization and system design