Nikhit Mago
Machine Learning Engineer, Special Projects at Amazon
Seattle, Washington, United States
Summary
👤
Senior🎓
Top SchoolNikhit Mago is a research-oriented Machine Learning Engineer with 8 years of experience building scalable ML systems across robotics, retail, and enterprise analytics. At Amazon he leads multi-node deep learning infrastructure work—enabling 16x larger training runs and fine-tuning massive MoE models on 128 H100 GPUs—while mentoring engineers on RL-driven agentic workflows. He combines deep expertise in distributed training (4D parallelism, activation recompute, offloading) with hands-on data engineering for petabyte-scale multimodal pipelines and edge deployment. Prior roles include driving production vision and sensor-fusion models for Just Walk Out stores and building high-impact NLP and information-extraction systems that slashed annotation and operational costs. Comfortable bridging research and production, he pairs SoTA model development with pragmatic engineering to deliver measurable business outcomes. Based in Seattle, he brings a rare mix of low-level systems optimization and applied ML product delivery.
8 years of coding experience
4 years of employment as a software developer
Master of Science, Applied Data Science, Master of Science, Applied Data Science at University of Southern CaliforniaBachelor of Technology, Computer Science and Engineering, Bachelor of Technology, Computer Science and Engineering at Manipal Institute of TechnologySt.Columba's School, New Delhi
English, Hindi
Github Skills (23)
asr8
speech-synthesis8
machine-translation7
automatic-speech-recognition7
speech7
pyspark7
speaker-recognition7
transformer-models6
generative-ai6
large-language-models6
spark6
text-to-speech6
ai5
multimodal5
machine-learning5
Programming languages (2)
Github contributions (5)
nikhitmago/lookalike-modelling
Aug 2018 - Aug 2018
Finding customer lookalikes using Machine Learning in PySpark
Contributions:8 commits, 7 pushes, 1 branch in 14 days
machine-learningpysparkspark
nikhitmago/time-series-classification-of-human-activities
Sep 2018 - Sep 2018
Contributions:5 commits, 3 pushes, 1 branch in 1 day
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.