Senior AI Training Infra Engineer - Distributed PyTorch

Genesis AI seeks a senior ML infrastructure engineer to optimize distributed training pipelines and systems on multi-node GPU clusters in London. You will profile bottlenecks, implement low-level kernels, and integrate CUDA/cuDNN/Triton into high-level frameworks to maximize throughput and utilization.

The role requires deep expertise in distributed systems, Python, and PyTorch with a track record of hardware-aware optimizations across CPU/GPU boundaries and data movement.

#J-18808-Ljbffr

Full job description from WhatJobs. Apply for this role →