Rohit Mehra - Machine Learning Engineer Ll at Amazon

Rohit Mehra

Machine Learning Engineer Ll at Amazon

Sunnyvale, California, United States

Join Prog.AI to see contacts

Summary

👤

Senior

🎓

Top School

Rohit 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.

9 years of coding experience

4 years of employment as a software developer

University of Texas at San Antonio

Bachelor’s Degree Computer Engineering, Bachelor’s Degree Computer Engineering at G. B. Pant University Of Agriculture & Technology

Nanodegree Artificial Intelligence, Nanodegree Artificial Intelligence at Udacity

High 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 (37)

face-recognition10

kafka-streams9

kafka-producer9

torcs9

kafka9

object-detection9

stream-processing8

keras-neural-networks8

kafka-consumer8

deterministic7

api6

yahoo6

simplify6

capsnet5

keras5

Programming languages (3)

JavaJupyter NotebookPython

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

face-recognitionkafkakafka-consumerkafka-producerkafka-streams

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.

Request Free Trial