Software Engineer (Python) - ML Tooling
Posted Updated 4
views
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