Mohit Khatwani
Summary
🤩
RockstarMohit Khatwani is a software engineer in Seattle with 3 years of industry experience building production ML systems and a strong research background from an MS in Computer Science at UMBC. He has applied ML and reinforcement learning across domains—fraud detection, location analytics, and brain-computer interfaces—moving models from prototyping in Python and PyTorch/Keras to production at companies including TransUnion, Grainger, and Google. Comfortable across languages (Python, C++, Java, Go) and distributed tooling (pySpark, GraphX), he pairs algorithmic depth (DQN, policy gradients, VAE, binarized CNNs) with practical engineering like C++ preprocessing for large datasets. Notably, his research work emphasizes reducing RL training time with human-in-the-loop approaches, reflecting a focus on efficient, usable AI rather than only raw model accuracy.
3 years of coding experience
Github Skills (30)
pytorch10
large-language-models10
llm10
deep-learning10
fine-tuning10
e2e10
mistral10
xla10
gpt10
jax10
post-training9
benchmarking9
compiler9
google-cloud8
gpu8
Programming languages (4)
Github contributions (5)
khatwanimohit/maxtext
Dec 2023 - Nov 2025
A simple, performant and scalable Jax LLM!
Contributions:5 pushes, 1 branch in 1 year 11 months
khatwanimohit/xla
Apr 2023 - May 2023
Enabling PyTorch on Google TPU
Contributions:6 pushes, 1 branch in 9 days
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.