Job Description
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
Find more English Speaking Jobs in United Kingdom on Arbeitnow
Similar Jobs
Senior Product Engineer (Agent Training and Evals)
prolific • Remote
Fixed-Term Field Sales & Training Executive — Sephora UK
Yepoda • London
Field Sales & Training Executive (CDD) — Sephora France
Yepoda • Paris
Field Sales & Training Executive (CDD) — Sephora France
Yepoda • Paris
Training Manager ProAV (m/w/d)
PureLink GmbH • Rheine
Fixed-Term Field Sales & Training Executive — Sephora UK
Yepoda • London
How to Apply
About Genesis
Genesis
London
Frequently Asked Questions
Where is the Training / AI Infrastructure position at Genesis located?
The Training / AI Infrastructure role at Genesis is based in London.
What type of employment is the Training / AI Infrastructure role at Genesis?
This Training / AI Infrastructure position is offered as Full Time.
What skills are needed for the Training / AI Infrastructure role at Genesis?
Key skills and focus areas for this role include Engineering & Research.
How do I apply for the Training / AI Infrastructure position at Genesis?
You can apply for the Training / AI Infrastructure role at Genesis directly from this page. Create a professional, ATS-ready resume with Clever CV to strengthen your application before you apply.