Senior ML Research Developer

Our client is a non-profit organization committed to advancing research and creating technical solutions that enable safe-by-design AI systems.

We are seeking a Senior Machine Learning Research Developer to join a highly technical team working at the forefront of machine learning research and development.

In this role, you will work closely with ML research scientists and engineers to solve challenging training and inference problems involving large-scale models. You will help translate research concepts into robust implementations, develop the infrastructure required to run complex experiments and optimise workflows for high-performance computing environments.

This is an opportunity to work at the intersection of machine learning research, software engineering and high-performance computing, contributing directly to the development and evaluation of advanced AI systems.

Key Responsibilities

• Collaborate closely with ML researchers to accelerate research, model training and inference.
• Facilitate the efficient implementation of novel machine learning models, architectures and simulated environments.
• Design and implement workflows for research experiments, ranging from small-scale experimental scenarios to large-scale projects.
• Develop tools, libraries and infrastructure to optimise the use of computing resources.
• Improve the efficiency, scalability and reproducibility of machine learning development workflows.
• Help translate research concepts and experimental ideas into reliable, production-quality implementations.
• Establish, document and maintain best practices for machine learning model development and experimentation.
• Identify technical bottlenecks across training and inference workflows and develop solutions to address them.
• Collaborate across research, engineering and infrastructure teams to support evolving technical requirements.

Background & Experience

• Degree in a relevant computer science or engineering discipline, such as Computer Science, Computer Engineering or Software Engineering.
• Advanced degree (MSc or higher) in Machine Learning, Computer Science or a related field, or equivalent professional experience.
• 5+ years of industry experience designing and implementing complex machine learning workflows.
• Strong hands-on experience working with high-performance computing environments and large-scale machine learning workloads.
• Extensive experience with one or more major ML frameworks, such as PyTorch, TensorFlow or JAX.
• Demonstrated experience contributing to high-quality research projects in deep learning.
• Experience working with large-scale model training and/or inference.

Technical Skills

• Experience working with cloud platforms such as AWS, GCP or Azure.
• Familiarity with containerisation technologies such as Docker and Kubernetes.
• Familiarity with modern data infrastructure and platforms, including technologies such as vector databases.
• Strong software engineering practices, including writing maintainable, well-tested and documented code.
• Understanding of distributed computing and high-performance ML workloads.

Collaboration & Communication

• Ability to collaborate effectively with research, engineering and infrastructure teams.
• Strong communication skills and the ability to translate complex research requirements into practical technical solutions.
• Ability to document technical approaches and establish effective engineering best practices.
• Demonstrated curiosity and willingness to stay current with developments in machine learning, deep learning and software engineering.

What You’ll Bring
We are looking for a technically strong ML engineer who is comfortable working in research-driven and highly experimental environments.
You should be equally comfortable engaging with research scientists on novel ideas and building the robust software, infrastructure and workflows required to turn those ideas into working experiments at scale.