Software Engineer (Python) - ML Tooling

Russell Tobin; Cambridge; Hybrid; GBP 70,000 - 90,000; Full time

Job summary

A technology staffing firm is seeking a Software Engineer IV (Python) to build and maintain ML tooling and infrastructure in Cambridge. This hybrid role involves collaborating with research teams to create scalable ML pipelines and develop systems connecting compute clusters. Candidates should have over 4 years of experience in software engineering, particularly with Python and frameworks like PyTorch. The position offers a dynamic environment focused on next-generation AI solutions.

Qualifications

• 4+ years industry experience as a Software Engineer with a focus on Python.
• Hands-on experience with PyTorch or similar deep learning frameworks.
• Experience building ML pipelines and training infrastructure.

Responsibilities

• Build robust, scalable ML tooling and infrastructure.
• Develop systems that connect with compute clusters.
• Work closely with researchers to translate needs into tooling.

Skills

Python

PyTorch

ML pipelines

Distributed compute

CI/CD

Tools

Compute clusters

Data storage

Kubernetes

Job description

Job Title: Software Engineer IV (Python) – ML Tooling & Infrastructure

Contract Duration: 12 months

Location: Cambridge, UK (Hybrid: 3 days onsite)

About the role

We are hiring a Python-focused Software Engineer to build and maintain machine learning tooling and infrastructure used by research teams working on next-generation AI experiences.

This role is ideal for engineers who enjoy building scalable ML pipelines, working with distributed compute, and collaborating directly with research scientists to deliver reliable internal tools.

What you’ll do

• Build robust, scalable ML tooling and infrastructure to support research workflows
• Develop systems that connect with compute clusters, data storage, source control, CI/CD
• Work closely with researchers to understand needs and translate them into stable tooling
• Own end-to-end engineering tasks: designing, coding, testing, deployment, support
• Debug complex issues across systems (data, training jobs, infrastructure, tooling)
• Contribute to sprint-based delivery: improvements, bug fixes, stability initiatives

What we’re looking for (must-have)

• 4+ years industry experience as a Software Engineer (Python-focused)
• Hands-on experience with PyTorch (or TensorFlow / similar DL frameworks)
• Experience building ML pipelines / training infrastructure / research tooling
• Practical exposure to distributed compute (e.g., Ray, Spark, Kubernetes, Slurm, etc.)
• Experience with CI/CD and deployment workflows

Nice-to-have

• Experience with audio ML, multimodal ML, or DSP-related ML
• Experience handling large real-world datasets, capture, or annotation pipelines
• Linux or shell scripting experience
• Exposure to hardware/prototype workflows or on-device deployment