Gaurav Jain
Software Engineer at Google
Los Altos, California, United States
Summary
🤩
Rockstar🎓
Top SchoolGaurav Jain is a seasoned software engineer with 14 years of experience building high-throughput systems, specializing in concurrency, algorithms, and deep learning from his base in Los Altos. At Google he works on Machine Perception within Research and Machine Intelligence and has driven critical infrastructure improvements across search, ads, and consumer apps—designing ad selection/formatting and optimizing cache and pipeline performance for large-scale services. He has hands-on experience with systems that handle >1M QPS and a strong background in ad serving, search infra, and scaling consumer backends. His open-source contributions span code analysis (improving language parsing in ScanCode), systems monitoring (Diamond), and TensorFlow ecosystem projects (TF-Agents and TensorFlow Probability), reflecting both backend/DevOps rigor and ML compatibility work. A graduate of IIT Delhi and UPenn (MS CS), he combines academic depth with practical product-scale engineering and even holds a patent from early research at IBM Watson Labs. Notably, he navigates both low-level performance tuning and modern ML stack migrations, bridging legacy systems and TF2-era best practices.
14 years of coding experience
1 year of employment as a software developer
Indian Institute of Technology Delhi (IIT Delhi)MS, Computer Science, MS, Computer Science at University of Pennsylvania
Github Skills (34)
probabilistic-programming10
ai-agent10
lexer10
python10
groovy10
machine-learning10
reinforcement-learning10
java10
regular-expression10
javas10
agent10
pygments10
tensorflow10
testing9
debug9
Programming languages (12)
Github contributions (5)
python-diamond/Diamond
Jan 2015 - Nov 2016
Diamond is a python daemon that collects system metrics and publishes them to Graphite (and others). It is capable of collecting cpu, memory, network, i/o, load and disk metrics. Additionally, it features an API for implementing custom collectors for gathering metrics from almost any source.
Role in this project:
Backend & DevOps Engineer
Backend & DevOps EngineerContributions:122 commits, 93 PRs, 81 pushes in 1 year 11 months
Contributions summary:Gaurav primarily focused on code style improvements and bug fixes within the Diamond project. They addressed PEP8 issues, corrected code errors in various collectors, and resolved issues such as reporting None values. Their work included updates to numerous collectors, including adjustments in data collection and minor fixes to several files. This suggests a strong understanding of the codebase and a focus on maintaining code quality and functionality.
memorypythonprometheusdiamondmonitoring
tensorflow/agents
Jan 2019 - Aug 2019
TF-Agents: A reliable, scalable and easy to use TensorFlow library for Contextual Bandits and Reinforcement Learning.
Role in this project:Back-end Developer & ML Engineer
Back-end Developer & ML EngineerContributions:22 commits in 6 months
Contributions summary:Gaurav primarily focused on addressing compatibility issues with TensorFlow's `TensorShapeV2` within the TF-Agents library, ensuring proper shape handling across different environments. They also made changes to remove reliance on deprecated session and placeholder usage within tests, aligning the project with modern TensorFlow practices. Furthermore, the user's commits touched upon various core functionalities of the library related to reinforcement learning algorithms and related tools, such as trajectory replay and PPO, indicating a strong understanding of the project's core domain.
scalabletf-agentsmultiagent-reinforcement-learningtensorflow-librarymulti-armed-bandits
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.