Training / AI Infrastructure
Genesis Therapeutics · London
IT & SoftwareCompany career page
The apply link worked recently (not employer-confirmed).
Role summary
Genesis Therapeutics seeks an AI Infrastructure engineer to optimize foundation model training systems. The role focuses on eliminating performance bottlenecks across the training stack, designing distributed systems for multi-node GPU clusters, and implementing low-level optimizations to improve convergence speed and hardware efficiency.
What they're looking for
- 8+ years experience in distributed systems, ML infrastructure, or high-performance computing
- Production-grade Python expertise
- Deep proficiency in CUDA, cuDNN, Triton, and custom kernel optimization
- Proven experience with PyTorch distributed training at scale
- System-level optimization expertise: CPU/GPU balance, memory management, data throughput
- Track record tuning hardware-software interactions for maximum utilization
What you'll be doing
- Profile and eliminate bottlenecks across foundation model training stack from data pipelines to GPU kernels
- Design and build distributed PyTorch training systems for multi-node GPU clusters with focus on scalability and robustness
- Implement efficient low-level code and integrate it into high-level training frameworks
- Develop monitoring and debugging tools to diagnose performance regressions in large-scale training runs
Privacy & handoff
This role links to Genesis Therapeutics's own application system. RoleLantern records your interest and checks the link, but the application happens on the employer's site and status updates may not come back to us. Clicking apply does not mark you as applied automatically.