Shauheen Zahirazami
Senior ML Engineering Manager at PyTorch
Seattle, Washington, United States
Summary
🤩
Rockstar🎓
Top SchoolShauheen Zahirazami is a Senior ML Engineering Manager with a PhD in electrical engineering and over 15 years of programming and technical management experience, currently leading ML infrastructure and accelerator tooling at AWS. He has driven cloud accelerator, framework and ecosystem work across Google, Microsoft and AWS—spanning TPU/TPU-like inference runtimes, Trainium/Neuron tooling, ONNX/WinML integrations, and ML.NET performance benchmarking. Known for blending control-systems rigor and applied mathematics with pragmatic software delivery, he excels at performance profiling, cluster management and AI-assisted optimization for compute stacks. A frequent contributor to foundation governance (PyTorch TAC/board roles) and creator of accelerator-focused developer tools, he bridges research, open-source ecosystems and production ML at scale. Based in Seattle, he combines deep quantitative modelling and systems design with a track record of shipping tooling that tightens the loop between ML research and deployment.
11 years of coding experience
25 years of employment as a software developer
PhD Electrical Engineering Control, PhD Electrical Engineering Control at Concordia UniversityUniversité de MontréalBSc Applied Mathematics, BSc Applied Mathematics at University of Tehran
French, English
Stackoverflow
1reputation
152reached
0answers
1question
Github Skills (38)
algorithms10
pytorch10
interoperability10
python10
mxnet10
acceleration10
machine-learning10
dotnet10
cross-platform10
onnx10
ml10
numpy10
keras10
deep-learning10
tensorflow10
Programming languages (6)
Github contributions (5)
shauheen/xla
May 2022 - Mar 2025
Enabling PyTorch on Google TPU
Contributions:4 pushes in 2 years 10 months
pytorchtpu
shauheen/pytorch
Jul 2021 - Dec 2023
Tensors and Dynamic neural networks in Python with strong GPU acceleration
Contributions:5 pushes, 1 branch in 2 years 4 months
pythongpu-accelerationdeep-learninggpuacceleration
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.