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 (30)
algorithms10
pytorch10
artificial-intelligence10
python10
machine-learning10
dotnet10
onnx10
numpy10
ml10
keras10
deep-learning10
tensorflow10
gpu10
ai10
neural-network10
Programming languages (6)
Github contributions (5)
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
shauheen/xla
May 2022 - Jun 2025
Enabling PyTorch on Google TPU
Contributions:6 pushes in 3 years 2 months
pytorchtpu
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.