Rohit Mehra
Machine Learning Engineer Ll at Amazon
Sunnyvale, California, United States
Summary
👤
Senior🎓
Top SchoolRohit Mehra is a Machine Learning Engineer II based in Sunnyvale with nine years of experience building production-grade ML and distributed systems for large-scale products. At Amazon he develops MLOps and edge model hosting—combining SageMaker, ECS Fargate, DynamoDB and CI/CD to operationalize ASR and core transformer workloads. He previously delivered BERT-based NER, cross-modal retrieval, and full-stack distributed warehousing solutions at Staples, and has hands-on experience with Spark, PyTorch, TensorFlow, and Hugging Face Transformers. Rohit’s background spans research-driven model innovation (CapsNet, UNet variants, FCNs) to pragmatic cloud-native deployments, giving him fluency across prototype-to-production lifecycles. He’s skilled in MLOps tooling (Docker, Kubernetes, MLflow) and large-scale data engineering, and often bridges research ideas with scalable engineering constraints. An interesting thread through his career is applying advanced NLP and computer-vision research to real-world e-commerce and edge-AI challenges.
10 years of coding experience
4 years of employment as a software developer
University of Texas at San AntonioBachelor’s Degree Computer Engineering, Bachelor’s Degree Computer Engineering at G. B. Pant University Of Agriculture & TechnologyNanodegree Artificial Intelligence, Nanodegree Artificial Intelligence at UdacityHigh School High School/Secondary Diplomas and Certificates, High School High School/Secondary Diplomas and Certificates at St. Joseph's Academy Dehradun
English, Hindi
Github Skills (27)
kafka-streams10
kafka-python10
face-recognition10
kafka-consumer10
kafka-producer9
torcs9
kafka9
stream-processing9
object-detection9
dynamic-routing7
capsnet7
vector5
keras5
spark4
kaggle4
Programming languages (3)
Github contributions (5)
rohit-mehra/boondh
Jul 2020 - Jan 2023
Contributions:2 releases, 27 pushes, 1 branch in 2 years 6 months
rohit-mehra/eye_of_sauron
Sep 2018 - Jul 2020
Stream processing using kafka-python to track people (user input images of target) in the wild over multiple video streams.
Contributions:72 commits, 25 pushes, 1 branch in 1 year 10 months
kafka-pythonstream-processingface-recognitionkafkaobject-detection
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.