Kunjan Patel
Machine Learning Engineer at Google
Seattle, Washington, United States
Summary
👤
Senior🎓
Top SchoolKunjan Patel is a Machine Learning Engineer with nine years of experience building high-performance ML infrastructure and production systems, currently focused on optimizing training and inference for diffusion models on XLA devices (TPU and GPU) using JAX. At Google he has written low-level TPU kernels and driven core ML performance improvements, and prior roles include optimizing distributed LLM inference on Kubernetes for GKE. He maintains the maxdiffusion project, demonstrating practical open-source work that accelerates diffusion model workflows on accelerators. Based in Seattle and trained at UCLA in Computer Science, he blends systems engineering rigor with ML research tooling. Colleagues describe him as a kernel-minded engineer who “plays games in his programs,” reflecting a playful curiosity that surfaces in efficient, elegant implementations.
9 years of coding experience
6 years of employment as a software developer
University of California, Los Angeles
Github Skills (58)
charts10
datatable10
devtools10
jsdoc10
datagrid10
node-server10
angular10
ui-components9
expressjs9
angular-ui9
javascript9
firefox-addon9
chrome-extension9
typescript9
generator9
Programming languages (3)
Github contributions (5)
kunjan343/TypeNode
Sep 2018 - Feb 2022
Advance typescript node server totally flexible with customization for enterprise applications. No overhead, No unnecessary customization, No custom guideline. Follow your own standards with simplicity.
Contributions:221 commits, 151 PRs, 155 pushes in 3 years 5 months
followexpressenterprise-applicationstypescript-frameworkhttp-server
kunjan343/kunjan343
Mar 2021 - Mar 2024
Contributions:7 pushes in 2 years 11 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.