AI Platform Support Engineer (APAC)
Lightning AI12 hours ago
Singapore, SingaporeMid Level
Responsibilities
- Partner directly with customer engineering teams running production ML training and inference workloads.
- Diagnose distributed systems, Kubernetes, GPU, PyTorch, networking, storage, inference, and platform reliability issues.
- Analyze logs, metrics, traces, and system behavior to identify root causes and performance bottlenecks.
- Advise customers during high-impact incidents and platform degradation events.
- Support workloads scaling across multi-node GPU systems and containerized or bare-metal environments.
- Drive reliability improvements through incident reviews, automation, tooling, documentation, runbooks, and improved observability.
- Collaborate with infrastructure, networking, and platform engineering teams to improve troubleshooting workflows and customer experience.
Requirements
- Strong software engineering and systems troubleshooting background.
- Experience with Kubernetes, containerized environments, cloud infrastructure, and distributed systems.
- Linux knowledge covering networking, storage, process management, and performance tuning.
- Experience using observability and debugging tools such as Prometheus, Grafana, or OpenTelemetry.
- Hands-on experience operating machine learning workloads in production or research environments.
- Experience with distributed ML systems and tooling such as PyTorch, CUDA, or NCCL.
- Familiarity with GPU infrastructure and orchestration, including troubleshooting ML infrastructure performance, reliability, and scaling issues.
- Strong communication skills and ability to work directly with technical customers and engineering teams.
- Preferred experience includes large-scale model training, distributed inference, Ray, Kubeflow, Slurm, InfiniBand, RDMA, high-performance networking, bare-metal infrastructure, ML storage systems, AI infrastructure, cloud, MLOps, developer tooling, platform engineering, operational tooling, and Python automation or scripting.
Benefits
- Medical, dental, and vision coverage for employees and eligible dependents.
- RSUs and location-dependent retirement or pension contributions.
- Unlimited PTO, company holidays, floating holidays, and a two-week company-wide winter break.
- Paid parental and family leave.
- Annual learning and development allowance.
- Wellness and work-from-home stipends.
- Four weeks of paid sabbatical after four years of service.
- Remote work for candidates in the Philippines or Singapore; office-based teams may use a hybrid model with flexible schedules.
- Sunday-Wednesday schedule from 7:00 AM to 5:00 PM local time in UTC+8.
- Complimentary meals at office hubs; benefits may vary by location, team, and role.
Tech Stack
Categories
About Lightning AI
The AI development platform - From idea to AI, Lightning fast ⚡️. Code together. Prototype. Train on GPUs. Scale. Serve. From your browser - with zero setup. AI Studio is your laptop on the cloud. Zero setup. Always ready. Persistent storage and environments. Code on CPU. Debug on GPU. Scale to multi-node. Run sweeps, jobs and more. Scale models with PyTorch Lightning, Fabric, Lit-GPT, torchmetrics and more.