Qualcomm AISW Hyderabad
Dec 2024 to present
Machine Learning Engineer
- Optimize and enable complex neural networks, including LLMs such as the Llama 3.1 family (3B to 8B) and GPT-3, for on-device inference on Snapdragon chipsets and NPUs, across CV, NLP, GenAI and multimodal models.
- Apply the full range of quantization techniques, including post-training and quantization-aware approaches, alongside accuracy tuning and system-level optimization of latency, memory and power.
- Work across the QNN SDK, QAIRT Converter and ONNX Execution Provider toolchains, and led priority delivery initiatives on the ONNX EP and QNN-QAIRT Converter workstreams.
- Apply LLM-specific optimizations: KV-cache quantization (INT8/FP8), weight-only quantization (W4A16/INT4), attention-head optimizations and runtime memory footprint reduction.
Jun 2023 to Dec 2024
Machine Learning Engineer, Associate
- Built model conversion pipelines and enabled early-stage neural networks on Qualcomm's AI Engine, across transformer and sequence-based architectures.
- Enabled deployment of CV, NLP and LLM workloads through quantization, model conversion, debugging and customer enablement.
Nov 2022 to Jun 2023
Interim Engineering Intern
- Wrote automation scripts for internal workflows and model conversion, working with TensorFlow, PyTorch and ONNX.