> Markdown version of [/jobs/ext/2238941-research-software-engineer-ml-training-infrastructure](https://www.wearedevelopers.com/jobs/ext/2238941-research-software-engineer-ml-training-infrastructure). Every page supports `.md` or `Accept: text/markdown`. Links point to the HTML versions so they work for humans too. Agent guide: [/agents.md](https://www.wearedevelopers.com/agents.md).

---

# Research Software Engineer - ML Training Infrastructure

- **Company:** Reflection
- **Location:** Greater London, UK
- **Contract:** Permanent contract
- **Skills:** Artificial Intelligence, Distributed Computing Environment, Distributed Systems, Machine Learning, Performance Tuning, Reinforcement Learning, Pytorch
- **Published:** August 26, 2026
- **Apply:** https://www.collegerecruiter.com/job/2815170203-research-software-engineer-ml-training-infrastructure

## About the Role

Reflection, based in the United Kingdom, is seeking a Software Engineer to architect and optimize the training infrastructure for AI models. The role focuses on building scalable systems for reinforcement learning and distributed training, requiring deep experience in distributed systems. Candidates should have practical skills in tools like PyTorch and JAX, and a robust understanding of performance optimization. The position offers top-tier compensation and comprehensive health benefits, alongside a supportive work environment.
#J-18808-Ljbffr

## Description

Reflection, based in the United Kingdom, is seeking a Software Engineer to architect and optimize the training infrastructure for AI models. The role focuses on building scalable systems for reinforcement learning and distributed training, requiring deep experience in distributed systems. Candidates should have practical skills in tools like PyTorch and JAX, and a robust understanding of performance optimization. The position offers top-tier compensation and comprehensive health benefits, alongside a supportive work environment.


## Related Videos

- [Introduction to Azure Machine Learning](https://www.wearedevelopers.com/videos/368-introduction-to-azure-machine-learning)
- [How AI Models Get Smarter](https://www.wearedevelopers.com/videos/1374-how-ai-models-get-smarter)
- [Practical performance tuning for Serverless Java on AWS](https://www.wearedevelopers.com/videos/2075-practical-performance-tuning-for-serverless-java-on-aws)
- [Photonic Computing: Programming a New Class of AI Accelerators (incl. Live Coding)](https://www.wearedevelopers.com/videos/100196-photonic-computing-programming-a-new-class-of-ai-accelerators-incl-live-coding)
- [Tour de Force: Open-Source LLM Inference Optimization from Simple to Sophisticated](https://www.wearedevelopers.com/videos/100099-tour-de-force-open-source-llm-inference-optimization-from-simple-to-sophisticated)
- [Getting Started with Machine Learning](https://www.wearedevelopers.com/videos/260-getting-started-with-machine-learning)

## Related Articles

- [How to Become an AI Engineer](https://www.wearedevelopers.com/magazine/331-how-to-become-an-ai-engineer)
- [What Are Large Language Models?](https://www.wearedevelopers.com/magazine/304-what-are-large-language-models)
- [MLOps And AI Driven Development](https://www.wearedevelopers.com/magazine/82-mlops-and-ai-driven-development)
- [MLOps – What’s the deal behind it?](https://www.wearedevelopers.com/magazine/125-mlops-what-s-the-deal-behind-it)
- [Coffee with Developers - Maria Apazoglou - Making AI understandable for all in production](https://www.wearedevelopers.com/magazine/475-coffee-with-developers-maria-apazoglou-making-ai-understandable-for-all-in-production)
- [Graph and AI Trends 2026: Why Is AI Running but Not Yet Delivering?](https://www.wearedevelopers.com/magazine/680-graph-and-ai-trends-2026-why-is-ai-running-but-not-yet-delivering)