Skip to main content
AX
Looking for a job

Alex Xiao

@alexxiao

Senior Machine Learning Systems Engineer at Tesla, scaling multimodal training and building recoverable GPU workflows for production models.

United States
Message

At Tesla, I led training and productionization for an end-to-end multimodal foundation-model workstream during the Cortex/V13 scale-up. I re-architected distributed training so multi-day runs could recover after worker or GPU failures instead of restarting from a full snapshot.

I also turned bespoke training scripts into reproducible Docker and Kubernetes launches with Argo Workflows and MLflow lineage. My release path connected offline regression, shadow validation, and staged rollout so deployments could be reproduced and rolled back to an exact known state.

At Tesla, I built a fleet-scale hard-example mining and auto-labeling system for FSD perception. It used embeddings, FAISS/ANN search, and uncertainty-based sampling to find rare production failures, then routed ambiguous sequences to human reviewers and high-confidence examples into training.

At Meta, I helped build Looper, a self-service real-time ML optimization platform, and built retrieval and ranking components for Instagram Suggested Posts. Earlier, I embedded a lightweight C++ classifier in Facebook's reactive-cache path, replacing manually maintained filtering rules with a continuously adapting prediction and retraining loop.

Experience

Work history, roles, and key accomplishments

Tesla logoTE
Current

Senior Machine Learning Scientist

Jan 2024 - Present (2 years 9 months)

Led training and productionization of an end-to-end multimodal foundation-model workstream during Tesla's Cortex/V13 scale-up, expanding to ~50K H100 GPUs and 4.2x more training data. Re-architected multi-node training with PyTorch FSDP/DDP, NCCL, and BF16 mixed precision, and built reproducible Docker/Kubernetes/Argo Workflows with MLflow lineage.

Tesla logoTE

Machine Learning Scientist

Jan 2022 - Dec 2023 (1 year 11 months)

Built a fleet-scale hard-example mining and auto-labeling system for FSD perception, turning rare production failures into targeted training cohorts. Developed Python/PySpark pipelines for filtering, embedding, and clustering, and introduced uncertainty-based sampling to prioritize unreliable clips.

Meta logoME

Senior Software Engineer

Jul 2021 - Dec 2021 (5 months)

Helped build Looper, Meta's self-service real-time ML optimization platform, supporting ~690 models across 90+ teams and ~4M predictions/sec. Designed declarative Python/PyTorch APIs and extended Ax/BoTorch optimization across architecture, features, and hyperparameters.

Education

Degrees, certifications, and relevant coursework

Carnegie Mellon University logoCU

Carnegie Mellon University

Bachelor of Science, Computer Science

2014 - 2018

Bachelor of Science in Computer Science with a minor in Machine Learning from Carnegie Mellon University, 2014-2018.

Interested in hiring Alex?

You can contact Alex and 90k+ other talented remote workers on Himalayas.

Message Alex

People also viewed

View all talent

Get matched with your dream remote job

Sign up now and join over 250,000+ remote workers who receive personalized job alerts, curated job matches, and more for free!

Sign up
Himalayas profile for an example user named Frankie Sullivan