WHAT YOU’LL DO
- Drive down wall-clock time to convergence by profiling and eliminating bottlenecks across the foundation model training stack stack, from data pipelines to GPU kernels
- Design, build, and optimize distributed training systems (PyTorch) for multi-node GPU clusters, ensuring scalability, robustness, and high utilization
- Implement efficient low-level code (CUDA, cuDNN, Triton, custom kernels) and integrate it seamlessly into high-level training frameworks
- Optimize workloads for hardware efficiency: CPU/GPU compute balance, memory management, data throughput, and networking
- Develop monitoring and debugging tools for large-scale runs, enabling rapid diagnosis of performance regressions and failures
WHAT YOU’LL BRING
- Deep experience in distributed systems, ML infrastructure, or high-performance computing (8+ years)
- Production-grade expertise in Python
- Low-level performance mastery: CUDA/cuDNN/Triton, CPU–GPU interactions, data movement, and kernel optimization
- Scaling at the frontier: experience with PyTorch and training jobs using data, context, pipeline, and model parallelism
- System-level mindset with a track record of tuning hardware–software interactions for maximum utilization
Vacancy posted 5 days ago
Similar jobs that could be interesting for youBased on the Training / AI Infrastructure in United Kingdom vacancy
- ...profiling and eliminating bottlenecks across the foundation model training stack stack, from data pipelines to GPU kernels - Design,... ...YOU’LL BRING - Deep experience in distributed systems, ML infrastructure, or high-performance computing (8+ years) - Production-grade...TrainingFull-time
£150k - £180k per annum
...plenty of growth opportunities? Join a rapidly growing cloud and AI infrastructure company helping enterprises modernise Kubernetes operations,... ...and direct exposure to large-scale GPU fabric and distributed training environments. The role also carries a mentorship remit over...TrainingPermanentHybrid workingRemoteFlexible hours- £69k - £89k per annumEstimated...Description YOU ARE As a Lead and Principal Infrastructure Architect, youownend-to-end responsibility for... ...compute infrastructure for large-scale AI and machine learning systems, including large-scale distributed training environments. You are the authority who translates...TrainingFull-time
- £65k - £87k per annumEstimated...About Nebius: Nebius is leading a new era in cloud infrastructure for the global AI economy. We are building a full-stack AI cloud platform that supports developers and enterprises from data and model training through to production deployment, without the cost and complexity...TrainingFull-time
- £55k - £74k per annumEstimated...About Nebius: Nebius is leading a new era in cloud infrastructure for the global AI economy. We are building a full-stack AI cloud platform that supports developers and enterprises from data and model training through to production deployment, without the cost and complexity...TrainingFull-time
- £102k - £136k per annumEstimated...experienced engineer with several years in infrastructure engineering who now takes on more complex... ...-impact work designing andoptimizingthe AI and machine learning infrastructure that... ...resources such as GPU clusters and distributed training environments, deploy AI systems and...TrainingFull-time
- £47k - £63k per annumEstimated...About Nebius: Nebius is leading a new era in cloud infrastructure for the global AI economy. We are building a full-stack AI cloud platform that supports developers and enterprises from data and model training through to production deployment, without the cost and complexity...Training
- £43k - £58k per annumEstimated...We are seeking a highly skilled Cloud Infrastructure Support Engineer to join our global Customer... ...Platform, High Performance Computing (HPC), and AI supercomputer fleet (some of the fastest... ...solutions that support seamless model training, deployment, and scaling. Join a team...TrainingFull-timeRotating shifts
- £108k - £142k per annumEstimated...packing, right-sizing), and think about GPU infrastructure as software to be engineered. A... ...programming in Rust or Go. About Together AI Together AI, the AI Native Cloud, is purpose... ...-level specialized models, and pre-training at massive scale for fully custom intelligence...Training
- £129k - £169k per annumEstimated...Looking to architect next-generation AI infrastructure that transforms engineering simulations? Join a world-class AI research laboratory... ...involves building and managing the infrastructure required to train complex simulation models on hundreds of GPUs, leading the orchestration...TrainingRemote jobPermanentVisa sponsorshipFlexible hours
- £46k - £62k per annumEstimated...About Nebius: Nebius is leading a new era in cloud infrastructure for the global AI economy. We are building a full-stack AI cloud platform that supports developers and enterprises from data and model training through to production deployment, without the cost and complexity...Training
- £83k - £108k per annumEstimated...Models , pushing the boundaries of generative AI, computer vision, and simulation. We are... ...are seeking a Machine Learning Systems & Infrastructure Engineer to build and own the systems that turn raw real-world data into trained world models and reliable production endpoints...Training
- £53k - £72k per annumEstimated...About Nebius: Nebius is leading a new era in cloud infrastructure for the global AI economy. We are building a full-stack AI cloud platform that supports developers and enterprises from data and model training through to production deployment, without the cost and complexity...TrainingFull-time
£110k - £130k per annum
...physical world. Be a part of it. Build a new generation of AI enabling robots to learn. They will pick up complex tasks... ...and a stake in the company. Take ownership of the infrastructure that powers the AI training. This is not a conventional infrastructure role. You will...TrainingPermanent£160k per annum
...Senior AI Infrastructure Leader | Pharmaceutical R&D We're supporting a global pharmaceutical organisation building a major AI function... ...-as-code , creating the foundations required to train, fine-tune and serve large-scale biomedical foundation models....TrainingPermanentFull-timeHybrid working£80k - £150k per annum
...Work on the cutting edge of AI, creating a platform to help change the future. This is an... ...challenges in the field. As a key member of the AI infrastructure team, you'll take ownership of the systems that power large-scale model training, architecting and optimising distributed...TrainingPermanent- £57k - £77k per annumEstimated...About Nebius: Nebius is leading a new era in cloud infrastructure for the global AI economy. We are building a full-stack AI cloud platform that supports developers and enterprises from data and model training through to production deployment, without the cost and complexity...Training
- £89k - £117k per annumEstimated...the following responsibilities: Design, build, and operate scalable, reliable, and secure cloud-native infrastructure supporting AI model development, training, evaluation, deployment, and inference across public cloud and hybrid environments. Partner with AI...TrainingApprenticeshipHybrid workingOn-siteWork from homeFlexible hours
£100k - £150k per annum
...ChatGPT moment. We're building a new kind of AI system that allows robots to learn... ...from a single demonstration. No months of training data collection. No painstaking programming... ...for an exceptional engineer to own the infrastructure powering our model training. This isn't...TrainingPermanent- £67k - £90k per annumEstimated...ABOUT ELEVENLABS ElevenLabs is an AI research and product company transforming how... ...stipend. ABOUT THE ROLE We’re hiring Infrastructure Engineers to build the platform every... ...team ships on. From the GPU clusters that train and serve our models to the cloud platform...TrainingFull-timeImmediate startRemote
$150k - $250k per annum
...the next step in your career? Join Mount Thor, a company building dedicated bare-metal Apple Silicon infrastructure to power next-generation workloads, including training data generation for computer-use models and managed agent deployments. This opportunity is ideal...TrainingPermanent£325k - £395k per annum
...create reliable, interpretable, and steerable AI systems. We want AI to be safe and... ...systems which power Safeguards, including infrastructure for data storage and management, metric and... ...an equivalent combination of education, training, and/or experience Required field of study...TrainingLong-term contractHybrid workingVisa sponsorshipOn-siteFlexible hours- £60k - £78k per annumEstimated...to grow across our digital, cloud, data, AI and cybersecurity capabilities, we're looking... ...capability portfolio, including cloud, infrastructure, applications, data & AI, cybersecurity,... ...continuous improvement, offering ample training programs, mentorship, and the chance to gain...TrainingLong-term contract
- £53k - £72k per annumEstimated...the forefront of applied research, engineering, infrastructure and deployment at scale. They will continue to scale their training to larger & more capable models. They will be... ...exists to be this company: to build a world where AI will be the engine behind economically valuable...TrainingFull-timeRemoteMonday to WednesdayFlexible hours
- £54k - £73k per annumEstimated...ABOUT ELEVENLABS ElevenLabs is an AI research and product company transforming how we interact with technology. We launched... ...co-working stipend. ABOUT THE ROLE Every model we train runs on infrastructure this role owns. We operate NVIDIA GPU clusters across bare...TrainingFull-timeImmediate startRemote
£325k - £390k per annum
...create reliable, interpretable, and steerable AI systems. We want AI to be safe and beneficial for... ...AI systems. About the role Anthropic's Infrastructure organization builds and operates the distributed systems that train, serve, and secure our AI models. That includes...TrainingLong-term contractHybrid workingVisa sponsorshipOn-siteFlexible hours£100k - £110k per annum
...looking for an exciting new opportunity? Join a fast-scaling UK AI infrastructure owner-operator building sovereign, renewable-powered GPU... ...rendezvous routing and GPU-to-CPU process affinity before large training workloads launch Write and maintain Infrastructure as Code...TrainingPermanent£140k - £160k per annum
...development? Join a sovereign cloud and AI platform that combines the openness of... ...private cloud, and high-performance GPU infrastructure with clear, predictable pricing, it helps... ...will design GPU supercomputing clusters for training and inferencing foundation models, turning...TrainingPermanentRemote£80k - £110k per annum
...Work on the cutting edge of AI. This is an opportunity to join an AI startup working... ...will architect and optimize distributed training across multiple GPUs and machines in AWS.... ...utilisation. We're looking for an experienced AI Infrastructure or MLOps Engineer who has already built...TrainingPermanent- £65k - £88k per annumEstimated...Who We Are Lightning AI is the company behind PyTorch Lightning. Founded in 2019, we... ...build an end-to-end platform for developing, training, and deploying AI systems—designed to... ...seeking an experienced Senior GPU & Compute Infrastructure Engineer to join our Infrastructure...TrainingLong-term contractFull-timeHybrid workingOn-siteRemote
Do you want to receive more vacancies?
Subscribe and receive similar vacancies to Training / AI Infrastructure. Be the first to apply!
