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
Senior AI Training Infra Engineer - Distributed PyTorch employer: Genesis AI
Genesis is an exceptional employer for those looking to make a significant impact in the AI and robotics field. With a competitive salary, meaningful equity, and a hybrid work model, employees enjoy a dynamic work culture that fosters innovation and collaboration. The opportunity to shape the recruiting function from the ground up, alongside industry leaders, ensures that team members can grow professionally while tackling some of the most challenging problems in technology today.
