Ruturaj Vaidya
Senior Member Of Technical Staff at AMD
Austin, Texas, United States
Summary
🤩
Rockstar🎓
Top SchoolRuturaj Vaidya is a software engineer and researcher with eight years of experience at the intersection of deep learning, compilers, reverse engineering, and software security. Currently a Member of Technical Staff at AMD, he works on JAX/XLA support and ROCm integration to enable high-performance ML on AMD GPUs, including float8 and sparse ops and PJRT/XLA FFI bring-up. His academic work at the University of Kansas produced publications and tooling for binary analysis and supply-chain security across ecosystems like npm and PyPI, and he previously built a static C++ call-stack profiler at Meta using LLVM passes. Ruturaj’s background spans systems engineering, kernel management, and compiler construction, giving him a rare blend of low-level systems expertise and ML-framework engineering. Based in Austin, he aims to be a leading expert in secure, high-performance machine learning frameworks and compilers.
8 years of coding experience
10 years of employment as a software developer
University of KansasBachelor's degree Electronics, Bachelor's degree Electronics at University of Mumbai
Marathi, Spanish, Hindi, English
Stackoverflow
2,056reputation
125kreached
77answers
33questions
Github Skills (18)
c-language10
python10
gpu-programming10
roc10
tensorflow10
xla10
jax10
cprogramming-language10
bazel9
clang8
cicd8
clangd8
machine-learning7
macos6
deep-learning6
Programming languages (12)
Github contributions (5)
jax-ml/jax
Mar 2024 - Nov 2025
Composable transformations of Python+NumPy programs: differentiate, vectorize, JIT to GPU/TPU, and more
Role in this project:
MLOps Engineer & Backend Developer
MLOps Engineer & Backend DeveloperContributions:22 reviews, 51 PRs, 82 comments in 1 year 8 months
Contributions summary:Ruturaj primarily focused on integrating and supporting ROCm, an AMD GPU platform, within the JAX framework. Their contributions included modifying build files to incorporate ROCm support, adding ROCm-specific checks, and implementing necessary changes to enable and test ROCm-based functionalities like float8 data type support and sparse matrix operations. They also worked on bring up pjrt support, and integrating with the XLA FFI.
gpujittpujax
Ruturaj4/pin_examples
Feb 2019 - Dec 2020
Contributions:32 pushes, 1 branch in 1 year 10 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.