---
title: "Training / AI Infrastructure"
company: "Genesis"
company_url: "https://www.remjobs.works/companies/genesis"
url: "https://www.remjobs.works/job/genesis-training-ai-infrastructure-65d8b7be-ca93-4d57-9cd8-53f5cb2415d5"
apply_url: "https://jobs.ashbyhq.com/genesis/2196f682-bd01-4f83-992d-367bbb3c8e8b"
workplace: hybrid
location: "London"
employment_type: full-time
seniority: mid
role: ai-machine-learning
region: united-kingdom
skills: ["python", "pytorch"]
date_posted: 2026-09-07T15:57:24.750Z
first_seen_by_remjobs: 2026-09-17T11:40:13.375Z
---

# Training / AI Infrastructure

**Genesis** · London

Apply: https://jobs.ashbyhq.com/genesis/2196f682-bd01-4f83-992d-367bbb3c8e8b

## About the role

##### 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

---

Source: Genesis's own career page, read by RemJobs. Canonical HTML version: https://www.remjobs.works/job/genesis-training-ai-infrastructure-65d8b7be-ca93-4d57-9cd8-53f5cb2415d5