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