Zachary Cain
Senior Software Engineer at Google
Seattle, Washington, United States
Summary
🤩
Rockstar🎓
Top SchoolZachary Cain is a Senior Software Engineer in Seattle with six years of specialized experience building machine vision and deep learning systems, and a longer track record at Google launching image classification, object detection, NLP, and multimodal models. He combines hands-on model development with ML infrastructure work—recently focusing on integrating TPUs and GPUs for public users to accelerate training pipelines. Zachary has contributed to the high-profile pytorch/xla project, improving test infrastructure, image processing, and metrics/logging for TPU-backed training. Previously he built recommendation and scheduling models at Timeful and large-scale data mining systems at LinkedIn, blending research-grade ML with production engineering. A Stanford CS graduate, he’s comfortable moving between model design, metric-driven evaluation, and low-level hardware integration to turn research ideas into deployable systems.
6 years of coding experience
2 years of employment as a software developer
Bachelor's degree, Computer Science, Bachelor's degree, Computer Science at Stanford University
English
Github Skills (10)
xla10
pytorch10
machine-learning10
test-automation10
python9
tensorboard9
image-recognition7
deep-learning7
compiler-compiler5
compiler5
Programming languages (6)
Github contributions (5)
pytorch/xla
Jul 2019 - Jul 2021
Enabling PyTorch on XLA Devices (e.g. Google TPU)
Role in this project:
ML Engineer & Test Automation Engineer
ML Engineer & Test Automation EngineerContributions:4 releases, 41 reviews, 168 commits in 1 year 11 months
Contributions summary:Zachary contributed to the `pytorch/xla` repository by implementing and modifying test infrastructure and image processing functionality for image recognition models. They added flags for test batch size, integrated TensorBoard support for logging metrics, and improved training update messages, adding examples/sec logging. The user also refactored code related to XLA metrics and learning rate scheduling.
pytorchxladeep-learningtpucompiler
zcain117/xla
Jul 2019 - Jul 2019
Contributions:6 pushes, 4 branches in 5 days
Find and Hire Top DevelopersWe’ve analyzed the programming source code of over 60 million software developers on GitHub and scored them by 50,000 skills. Sign-up on Prog,AI to search for software developers.