Abhijit Balaji
Senior Software Engineer, GenAI at Google
Greater Seattle Area United States
Summary
🤩
Rockstar🎓
Top SchoolAbhijit Balaji is a Senior Software Engineer focused on GenAI with eight years of experience building production-grade ML systems across on-device perception, multimodal retrieval, and LLM deployment. Currently on Google’s Gemini-CLI team, he builds AI agents and CLI tooling, and previously led ML efforts at Rain where he designed TorchMX, a PyTorch-to-hardware quantization framework and shipped an on-premise Copilot-style coding assistant. His background spans product-facing ML at Google and Adobe, research in computational photography and portrait relighting at Samsung, and practical end-to-end machine vision systems from an early startup. He contributes to open-source tooling—integrating TensorFlow object detection into Prodigy annotation workflows—and blends low-level model optimization with UX-minded deployment. Known for turning research ideas into production features, he combines deep learning, computer vision, and NLP expertise with a hands-on engineering approach.
9 years of coding experience
6 years of employment as a software developer
Master of Science - MS, Computer Science, Master of Science - MS, Computer Science at The University of Texas at DallasBachelor of Technology - BTech, Mechanical Engineering, Bachelor of Technology - BTech, Mechanical Engineering at Vellore Institute of Technology
English, Tamil, Hindi
Stackoverflow
1,920reputation
446kreached
18answers
25questions
Github Skills (19)
prodigy10
python10
annotations10
machine-learning10
annotator10
tensorflow10
object-detection10
natural-language-processing9
nlp9
computer-vision9
spacy8
dataannotations8
data-annotation8
active-learning7
keras6
Programming languages (9)
Github contributions (5)
explosion/prodigy-recipes
Jun 2019 - Jul 2019
🍳 Recipes for the Prodigy, our fully scriptable annotation tool
Role in this project:
ML Engineer
ML EngineerContributions:26 commits, 2 PRs, 7 comments in 29 days
Contributions summary:Abhijit focused on integrating a Tensorflow-based object detection model into the Prodigy annotation tool, creating recipes to load and use the model for image annotation. They implemented hooks for Tensorflow Serving to enable predictions and added functionality for processing images, extracting bounding boxes, and displaying them within the Prodigy interface. The contributions included refactoring code, adding time logging for performance measurement, and fixing bugs related to the model loading and serving.
nlprecipesactive-learningdata-annotationdata-science
Abhijit-2592/model-server
Jun 2019 - Oct 2020
gRPC server for hosting ML models trained on any framework in python
Contributions:4 releases, 43 commits, 4 PRs in 1 year 4 months
grpcmlmodelpythondeep-neural-networksmodel-deployment
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.