> Markdown version of [/jobs/ext/2245973-member-of-technical-staff-research-software-engineer](https://www.wearedevelopers.com/jobs/ext/2245973-member-of-technical-staff-research-software-engineer). 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). --- # Member of Technical Staff - Research Software Engineer - **Company:** Reflection - **Location:** Greater London, UK - **Contract:** Permanent contract - **Skills:** Artificial Intelligence, Data Deduplication, Data Infrastructure, Data Transformation, Distributed Computing Environment, Distributed Systems, Machine Learning, Remote Direct Memory Access, Tokenization, Reinforcement Learning, Graphics Processing Unit (GPU), High Performance Computing, Pytorch, Containerization, Kubernetes, Slurm, Data Pipelines - **Published:** August 26, 2026 - **Apply:** https://www.collegerecruiter.com/job/2815163781-member-of-technical-staff-research-software-engineer ## About the Role * You are a strong software engineer who speaks the language of machine learning. * You may not have a PhD, but you know how to implement a research paper. * You have deep experience in at least one of the following: Distributed Training & Inference or Data Infrastructure * You enjoy working at the boundary between: * Machine learning algorithms * Distributed systems * High-performance computing You care deeply about performance, numerical stability, and reproducibility. You thrive in high-agency environments and enjoy solving hard technical problems. ## Description Overview Reflection's mission is to build open superintelligence and make it accessible to all. We're developing open weight models for individuals, agents, enterprises, and even nation states. Our team of AI researchers and company builders come from DeepMind, OpenAI, Google Brain, Meta, Character.AI, Anthropic and beyond. Responsibilities Bridge the gap between research and production by turning cutting-edge algorithms into scalable training systems. You will design and optimize the core infrastructure behind frontier AI models - from reinforcement learning training loops and distributed GPU training to massive-scale data pipelines. Our systems train models across thousands of GPUs and process petabyte-scale datasets. We care deeply about numerical stability, throughput, and reproducibility. This team owns and evolves the core infrastructure behind our training systems. We Focus On * Reinforcement learning training infrastructure * Distributed training and inference systems * Experiment infrastructure and reproducibility * Large-scale data pipelines The goal is to build the engineering foundation that allows researchers to iterate quickly while training models at massive scale. About The Role You will architect and optimize the core training infrastructure that powers our models. This includes RL training loops, distributed GPU systems, and large-scale data pipelines. You will work closely with researchers to transform new ideas into reliable, scalable training systems. Responsibilities Include * Designing and optimizing large-scale training loops and data pipelines. * Implementing state-of-the-art techniques and ensuring they are numerically stable and computationally efficient. * Building internal tooling for launching, monitoring, and reproducing complex experiments. * Diagnosing deep bottlenecks across the training stack (GPU memory issues, communication overhead, dataloader stalls). * Translating research prototypes into reusable, production-grade infrastructure. What You\'ll Work With Distributed Training * GPU parallelism (data, tensor, pipeline, expert) * Large-scale distributed training infrastructure * Communication optimization (NCCL, RDMA, GPU interconnects) * FSDP / ZeRO and model sharding Orchestration & Runtime Systems * Ray, Kubernetes, Slurm * Distributed runtimes and async systems * Containerization and sandboxing Frameworks * PyTorch * JAX * Megatron-style training stacks * Triton / custom kernels Data Infrastructure * Large-scale dataset curation pipelines * Deduplication and filtering systems * Tokenization and preprocessing * Distributed data processing frameworks About You * You are a strong software engineer who speaks the language of machine learning. * You may not have a PhD, but you know how to implement a research paper. * You have deep experience in at least one of the following: Distributed Training & Inference or Data Infrastructure * You enjoy working at the boundary between: * Machine learning algorithms * Distributed systems * High-performance computing You care deeply about performance, numerical stability, and reproducibility. You thrive in high-agency environments and enjoy solving hard technical problems. What We Offer * Top-tier compensation: Salary and equity structured to recognize and retain the best talent globally. * Health & wellness: Comprehensive medical, dental, vision, life, and disability insurance. * Life & family: Fully paid parental leave for all new parents, including adoptive and surrogate journeys. Financial support for family planning. * Benefits & balance: paid time off when you need it, relocation support, and more perks that optimize your time. * Opportunities to connect with teammates: lunch and dinner are provided daily. We have regular off-sites and team celebrations. ## Related Videos - [Running Secure Life Science Research at Scale using Hybrid GPU HPC and Kubernetes 🧬](https://www.wearedevelopers.com/videos/100355-running-secure-life-science-research-at-scale-using-hybrid-gpu-hpc-and-kubernetes) - [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) - [Understanding Kubernetes in a visual way](https://www.wearedevelopers.com/videos/100085-understanding-kubernetes-in-a-visual-way) - [From Messy Queries to Scalable Systems - How Data Engineering actually works](https://www.wearedevelopers.com/videos/100203-from-messy-queries-to-scalable-systems-how-data-engineering-actually-works) - [Serverless deployment of (large) NLP models ](https://www.wearedevelopers.com/videos/158-serverless-deployment-of-large-nlp-models) - [Instant KAI Sandboxes with vCluster: Multi-Tenant, Multi-Scheduler GPU Sharing](https://www.wearedevelopers.com/videos/100333-instant-kai-sandboxes-with-vcluster-multi-tenant-multi-scheduler-gpu-sharing) ## Related Articles - [How We Built a Worry-Free System That Runs for 10+ Years – And What We’d Do Again](https://www.wearedevelopers.com/magazine/751-how-we-built-a-worry-free-system-that-runs-for-10-years-and-what-we-d-do-again) - [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) - [Dev Digest 120 - Apple and peers](https://www.wearedevelopers.com/magazine/455-dev-digest-120-apple-and-peers) - [Navigating the AI Shift](https://www.wearedevelopers.com/magazine/629-navigating-the-ai-shift) - [Résumé-Driven Development: How IT trends affect the job market for software developers](https://www.wearedevelopers.com/magazine/59-resume-driven-development-how-it-trends-affect-the-job-market-for-software-developers) - [How to Become an AI Engineer](https://www.wearedevelopers.com/magazine/331-how-to-become-an-ai-engineer)