AI Platform Engineer for Production ML & Kubernetes

United States Digital Space LLC; Greater London; On-site; GBP 75,000 - 95,000; Full time

Benefits offered by this job

Comprehensive health coverage

Equity (RSUs)

Hybrid work model

Flexible time off

Paid parental leave

Job summary

Lightning AI, the company behind PyTorch Lightning, is hiring AIPlatform Support Engineers for the EMEA Customer Experience team. You will partner with ML engineers, diagnose failures in distributed systems, and guide customers through complex Kubernetes and GPU workloads in production.

We seek engineers who can diagnose performance bottlenecks, improve observability, and build runbooks. This hybrid London role requires in-office presence at least 2 days/week and supports large-scale AI

Qualifications

• Strong software engineering and systems troubleshooting background.
• Experience with Kubernetes and containerized environments.
• Linux systems knowledge, including networking, storage, process management, and performance tuning.
• Experience with cloud infrastructure and distributed systems.
• Experience with observability and debugging tools such as Prometheus, Grafana, or OpenTelemetry.

Responsibilities

• Partner directly with customer engineering teams running training and inference workloads in production.
• Help customers diagnose and resolve complex distributed systems and ML infrastructure issues.
• Act as a technical advisor during high impact incidents and platform degradation events.
• Translate infrastructure level issues into actionable guidance for ML engineers.
• Build credibility with customers through strong technical reasoning and clear communication.
• Investigate failures involving distributed training, Kubernetes orchestration, GPU allocation, networking, and storage systems.
• Troubleshoot PyTorch, CUDA, NCCL, and inference serving related issues.
• Analyze logs, metrics, traces, and system behavior to isolate root causes.
• Debug containerized workloads running across Kubernetes and bare metal GPU environments
• Support customers scaling workloads across multi node GPU systems
• Diagnose performance bottlenecks involving compute, memory, networking, or storage

Skills

Kubernetes

Linux

Cloud infrastructure

Distributed systems

Observability

Tools

Prometheus

Grafana

OpenTelemetry

PyTorch

CUDA

NCCL

Job description

Lightning AI, the company behind PyTorch Lightning, is hiring AIPlatform Support Engineers for the EMEA Customer Experience team. You will partner with ML engineers, diagnose failures in distributed systems, and guide customers through complex Kubernetes and GPU workloads in production.

We seek engineers who can diagnose performance bottlenecks, improve observability, and build runbooks. This hybrid London role requires in-office presence at least 2 days/week and supports large-scale AI