Jevin Jiang
Summary
🤩
RockstarJevin Jiang is a Software Engineer III with nine years of experience building high-performance ML and cloud systems, currently working on XLA and TPU compiler optimizations at Google. He has a strong track record accelerating ML workloads—at Oracle he ported and optimized a multivariate anomaly detection algorithm to CUDA with a 1600x speedup and added multi-GPU support and memory estimation tooling. Prior roles span Play Store personalization, founding a startup that shipped multiple React/React-Native apps, and embedded/IoT proof-of-concepts at Tencent, showing comfort across full-stack, systems, and low-level code. Jevin holds a BS in Computer Science with high honors from Michigan State and an MS from Yale, blending rigorous academics with production impact. He leans toward performance-critical engineering and has repeatedly translated algorithms into scalable, production-ready implementations that remove practical limits (e.g., multi-GPU training and GPU memory estimation).
10 years of coding experience
Github Skills (7)
xla10
jax10
python10
tpu10
mlr10
cprogramming-language9
c-language9
Programming languages (2)
Github contributions (5)
jax-ml/jax
Sep 2023 - Mar 2026
Composable transformations of Python+NumPy programs: differentiate, vectorize, JIT to GPU/TPU, and more
Role in this project:
Back-end Developer
Back-end DeveloperContributions:1 comment in 2 years 5 months
Contributions summary:Jevin primarily contributed to the Mosaic component, focusing on implementing and enhancing the support for TPU operations within the JAX ecosystem. Their work included adding support for sublane strided load and store, which involved modifying existing code in `jaxlib/mosaic/dialect/tpu/transforms` and `jaxlib/mosaic/python/apply_vector_layout.py`. Furthermore, the user implemented various optimizations and added new features, such as support for relayout operations and improving the efficiency of memory operations. They also made improvements for the `scf.for` loop and the `tpu.bitcast` op.
gpujittpujax
by-the-w3i/resume
Sep 2016 - May 2018
Contributions:8 pushes, 1 branch in 1 year 7 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.