Software Engineer (Python) - ML Tooling

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

#J-18808-Ljbffr

Full job description from WhatJobs. Apply for this role →