Skip to main content
Pukhraj GuptaPG
Looking for a job

Pukhraj Gupta

@pukhrajgupta

Machine Learning Engineer at AMD optimizing ML models for NPUs and built evaluation infrastructure that cut turnaround by over 80%.

India
Message

At AMD, via TalTech, I work on Vitis AI/ML compiler flows that map machine-learning kernels and operators to NPU hardware on Strix and VEK embedded devices. I optimize model offloading, resource utilization, and execution readiness.

I quantize and lightweight architectures using methods including INT8, FP16, PTQ, and QAT, and analyze vision and transformer models for deployment. I also optimize YOLO-based detectors for latency and FPS under NPU constraints.

I built and maintained evaluation and automation infrastructure spanning 1,500+ models, with ONNX evaluation, memory/MAC analysis, and validity checks. This cut evaluation turnaround by over 80%.

I also build RAG, MCP servers, and plan-execute multi-agent tooling to expose internal ML and NPU tools to LLM agents. Earlier, as a Research Intern at Indian Institute of Foreign Trade, I summarized research on blockchain in supply chains and used AHP and DEMATEL for decision analysis.

Experience

Work history, roles, and key accomplishments

AMD logoAM
Current

Machine Learning Engineer

Jan 2024 - Present (2 years 8 months)

Work on AMD Vitis AI/ML compiler flows for mapping ML kernels to NPU hardware, improving model offload efficiency and execution readiness. Quantized and lightweight 50+ architectures, built large-scale evaluation infrastructure, and developed RAG and multi-agent tooling.

Education

Degrees, certifications, and relevant coursework

National Institute of Technology Hamirpur logoNH

National Institute of Technology Hamirpur

BTech, Mathematics and Computing

2020 - 2024

Grade: 8.05

Pursued a BTech in Mathematics and Computing with a GPA of 8.05.

Coursera logoCO

Coursera

Machine Learning Specialization, Machine Learning

Completed the Machine Learning Specialization covering supervised learning, advanced algorithms, unsupervised learning, recommenders, and reinforcement learning.

Coursera logoCO

Coursera

Deep Learning with PyTorch, Keras and TensorFlow Professional Certificate, Deep Learning

Completed the Deep Learning with PyTorch, Keras and TensorFlow Professional Certificate, including 4 courses and 1 capstone.

Interested in hiring Pukhraj?

You can contact Pukhraj and 90k+ other talented remote workers on Himalayas.

Message Pukhraj

People also viewed

View all talent

Get matched with your dream remote job

Sign up now and join over 250,000+ remote workers who receive personalized job alerts, curated job matches, and more for free!

Sign up
Himalayas profile for an example user named Frankie Sullivan