Software Engineer (Python) - ML Tooling
Russell Tobin
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
Reference: WJ-5107_13652713