Chirag Jain
Senior Machine Learning Engineer at TrueFoundry
Mumbai, Maharashtra, India
Summary
🤩
Rockstar🎓
Top SchoolChirag Jain is a Senior Machine Learning Engineer with 11 years of experience building scalable MLOps platforms and production-grade ML systems, currently designing cloud-agnostic tooling at TrueFoundry. He combines deep NLP research—authoring work presented at AAAI, EMNLP and ICON—with hands-on engineering that productionizes models across Kubernetes, GPUs, and multi-tenant deployments. His background at Haptik includes optimizing high-throughput conversational AI pipelines, memory-aware model serving, and orchestration for zero-downtime migrations. An active open-source contributor, he has improved notable projects like jalammar/ecco (model explainability) and nlpaug (augmentation), and often bridges research and engineering by adapting libraries to real-world models like T5. Known for pragmatic system design, he also enjoys competitive programming and maintaining readable, well-documented code across repos.
10 years of coding experience
6 years of employment as a software developer
High School, Computer Science, Distinction - 91%, High School, Computer Science, Distinction - 91% at Thakur College of Science and CommerceHigh School, Distinction - 94.55%, High School, Distinction - 94.55% at Jayaben B. Khot High SchoolBachelor's Degree, Computer Engineering, 8.1/10, Bachelor's Degree, Computer Engineering, 8.1/10 at Dwarkadas J. Sanghvi College of Engineering
English, Hindi, Marathi
Stackoverflow
764reputation
39kreached
19answers
6questions
Github Skills (37)
pytorch10
debug10
language-model10
named-entity-recognition10
python10
amazon-elasticsearch10
configuration-management10
machine-learning10
ml10
elasticsearchquery10
aws-elasticsearch10
natural-language-processing10
elasticsearch10
gpt10
nlp10
Programming languages (12)
Github contributions (5)
hellohaptik/chatbot_ner
Mar 2017 - Feb 2022
chatbot_ner: Named Entity Recognition for chatbots.
Role in this project:
Back-end Developer
Back-end DeveloperContributions:1 release, 65 reviews, 480 commits in 4 years 11 months
Contributions summary:Chirag's commits primarily focused on improving the codebase for Named Entity Recognition (NER) in chatbots. Their work involved fixing bugs within the date detection module, adding documentation for the data store and Elasticsearch classes, and refactoring the code. They also updated the installation and setup steps for the project, demonstrating a focus on maintainability and usability.
nlpchatbotsnamed-entity-recognitionnerentity-recognition
axolotl-ai-cloud/axolotl
Nov 2023 - Dec 2024
Go ahead and axolotl questions
Role in this project:Backend Developer
Backend DeveloperContributions:17 reviews, 16 PRs, 54 comments in 1 year 1 month
Contributions summary:Chirag primarily contributed to bug fixes and improvements within the axolotl project. Their work included fixing typos, updating configuration settings for machine learning models, adjusting logging configurations, and addressing issues related to early stopping during training. The user also made enhancements to the training flow, including handling model saving and plugin integrations. These changes focused on improving the stability, functionality, and overall usability of the axolotl framework.
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.