ELEthan Li.
Back to selected work

ML PLATFORM ENGINEERING

Deep Learning Training Platform

Machine Learning Platform Tech Lead

Architected a unified ML training platform to schedule workloads across multiple Kubernetes GPU clusters and led a team of approximately 10 engineers, collaborating with product managers and ML researchers.

KubernetesMulti-cluster SchedulingGPU TrainingAutomated Annotation
Unified training platform across Kubernetes clustersTraining workloads enter our internally developed training platform, which orchestrates jobs across multiple Kubernetes clusters through each cluster’s API. Each cluster’s scheduler places training pods on its GPU worker nodes. Two representative clusters are shown; the cluster count is illustrative.Training workloadsCode · data · resource requirementsINTERNALLY DEVELOPED SYSTEMDeep Learning Training PlatformUnified job submission · Cross-cluster orchestrationSchedule through each cluster’s Kubernetes APIKUBERNETES CLUSTER AKubernetes APITraining jobs / podsCluster schedulerGPU pod placementGPU worker nodesTraining podsTraining podsTraining podsKUBERNETES CLUSTER BKubernetes APITraining jobs / podsCluster schedulerGPU pod placementGPU worker nodesTraining podsTraining podsTraining pods…Representative clusters shown · Platform spans multiple clusters
Architecture overview · Our training platform orchestrates workloads across multiple Kubernetes clusters. Cluster count is illustrative.

01 /Platform architecture

Architected an internally developed, end-to-end machine learning training platform that sits in front of the Kubernetes APIs and provides unified scheduling across multiple GPU clusters. The platform orchestrates training workloads across clusters, while each cluster schedules training pods onto its GPU worker nodes.

02 /Technical leadership

Led a team of approximately 10 engineers, driving cross-functional collaboration with product managers and machine learning researchers.

03 /Automated annotation

Developed models for automated data annotation, accelerating model training and deployment.

04 /Training performance

Performed performance analysis on machine learning training jobs and resolved bottlenecks to improve training efficiency.

NEXT PROJECT

Engineering Efficiency Platform