AI Platform Support Engineer

About the role

Lightning AI is looking for an AI Platform Support Engineer to join the EMEA Customer Experience team. You will act as a technical partner to ML teams, helping them diagnose and resolve complex distributed systems and infrastructure issues in production environments. This role involves working directly with customers to improve reliability and performance across Kubernetes, GPU platforms, and distributed ML workloads.

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
• Investigate failures involving distributed training, Kubernetes orchestration, GPU allocation, networking, and storage systems
• Troubleshoot PyTorch, CUDA, NCCL, and inference serving related issues
• Build internal tooling, automation, documentation, and runbooks
• Identify recurring patterns across customer issues and drive long term reliability improvements

Required skills

• Software engineering
• Systems troubleshooting
• Kubernetes
• Linux
• Cloud infrastructure
• Distributed systems
• Prometheus
• Grafana
• OpenTelemetry
• Machine learning
• PyTorch
• CUDA
• NCCL
• GPU infrastructure

Nice to have

• Large scale model training
• Distributed inference systems
• Ray
• Kubeflow
• Slurm
• InfiniBand
• RDMA
• Python

Qualifications

• Strong software engineering and systems troubleshooting background
• Hands on experience operating machine learning workloads in production or research environments

Benefits

• Medical, dental, and vision coverage
• RSUs
• Pension contributions
• Unlimited PTO
• Company-wide winter break
• Paid parental and family leave
• Professional development allowance
• Wellness stipend
• Work-from-home stipend
• Sabbatical program

About the Company

Lightning AI is the company behind PyTorch Lightning, building an end-to-end platform for developing, training, and deploying AI systems. Through its merger with Voltage Park, the company combines developer-first software with cost-efficient, large-scale compute to serve researchers, startups, and large enterprises.