Sambhav Jain
Principal Engineer at AMD
San Francisco, California, United States
Summary
🤩
Rockstar🎓
Top SchoolSambhav Jain is a Principal Engineer with a decade of experience building and leading machine-learning compiler and acceleration efforts, now focused on making AMD GPUs fast and accessible. He has driven MLIR-based compiler architectures and tooling at Cruise—shipping an ML compiler frontend that onboarded a real model within six months—and previously led quantization-aware training and graph-transform frameworks at Xilinx that were adopted upstream in Vitis-AI and PyTorch. Sambhav blends deep research roots (Stanford MS, publications, and patents) with hands-on production expertise across compiler infra, CI/CD automation (notably improving Torch-MLIR build and CI workflows), and device-targeted optimizations for resource-constrained systems. He excels at turning algorithmic ideas into deployable compiler components and developer-friendly toolchains, with a track record of presenting work at venues like GTC and contributing to prominent open-source projects. Based in San Francisco, he pairs systems-level thinking with pragmatic engineering to accelerate ML model deployment across diverse hardware backends.
10 years of coding experience
10 years of employment as a software developer
Bachelor of Technology (B.Tech.) Electrical and Electronics Engineering, Bachelor of Technology (B.Tech.) Electrical and Electronics Engineering at National Institute of Technology, TiruchirappalliMaster of Science (M.S.) Electrical Engineering, Master of Science (M.S.) Electrical Engineering at Stanford University
Github Skills (14)
automation10
automations10
docker10
devops10
bazel10
dockers10
cicd10
clangd9
clang9
ubuntu9
compiler-compiler8
compiler8
mlr8
python4
Programming languages (11)
Github contributions (5)
llvm/torch-mlir
Jun 2022 - Dec 2022
The Torch-MLIR project aims to provide first class support from the PyTorch ecosystem to the MLIR ecosystem.
Role in this project:
DevOps Engineer & Automation Engineer
DevOps Engineer & Automation EngineerContributions:147 reviews, 26 commits, 148 PRs in 5 months
Contributions summary:Sambhav primarily focused on improving the build and CI/CD infrastructure of the project. Contributions included adding a Docker wrapper for local and CI builds, implementing Bazel caching for faster workflow execution, and updating the build environment to Ubuntu 22.04 and clang-16. The user also introduced a process to verify and commit changes from the Buildifier linter. This work streamlined the build process and improved development efficiency.
pytorchmlirtorchcompilerecosystem
cruise-automation/mlir-tcp
Oct 2023 - Jan 2025
Tensor Compute Primitives: Mid-level Intermediate Representation for Machine Learning Programs
Contributions:132 reviews, 120 PRs, 60 pushes in 1 year 3 months
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.