preloader

Hi

I am currently employed at Advanced Micro Devices, Inc. as a Member of Technical Staff (MTS), where I focus on developing scalable, robust kernels for Generative AI models optimized for AMD's AI Engine Block. Previously, I have worked at Qualcomm AI Research as a Senior ML Engineer and Amazon Lab126 Hardware Compute Group as a SysDE ML Compiler Engineer. I graduated with two research focussed masters from University of Southern California in 2020 and an integrated dual degree from Indian Institute of Technology, Kharagpur in 2017. At present, I am finishing up my doctorate from University of Texas, Austin where my research focus is designing energy-efficient, robust, machine learning algorithms for generative AI edge applications.

Feel free to check out my socials and connect with me on them. I am always open to new opportunities and collaborations.

I have a pet cat, Oliver. He is quite chonky 😄 and a very good boy. Check out his

about-me

EXPERIENCE

  • AMD Xilinx

    Member of Technical Staff | Feb 2024 - Present
    Austin, TX, USA

    Developing scalable, robust kernels for Generative AI models optimized for AMD's AI Engine Block.

  • Qualcomm AI Research

    Senior ML Engineer | Aug 2023 - Feb 2024
    Austin, TX, USA

    Worked as part of AI Research bringing Generative AI to the edge.

    Senior ML Engineer | Feb 2023 - Aug 2023
    New York City, NY, USA

    Worked as part of AI Compiler Labs group on improving the Hexagon Tensor Processor.

  • Amazon Lab126

    SysDE ML Compiler Engineer I | Sep 2020 - Nov 2022
    Sunnyvale, CA, USA

    Worked as part of Hardware Compute Group on the founding team for AZ1 Neural Edge Processor chip.

    Optimized software stack to deploy models on the edge.

  • Mediatek Research

    Research Intern | May 2020 - Aug 2020
    San Jose, CA, USA

    Worked on reducing training/inference computation complexity for RNN/LSTM running on accelerator architectures.

  • U. of Southern California

    Research Assistant | Aug 2017 - May 2020
    Los Angeles, CA, USA

    [HAL Group] Researched in Algorithm and Architecture co-design for Neural Networks Training on Accelerators.

    [Project SHARP] Analyzed graph theory problems - PageRank, NxContingency, graph embedding and belief propagation, etc.

    Teaching Assistant | Aug 2017 - Jul 2020
    Los Angeles, CA, USA

    Over the course of nine consecutive semesters, I instructed a total of nine classes at both the graduate and undergraduate levels.

  • Qualcomm India Pvt. Ltd.

    Systems Engineering Intern | May 2016 - Jul 2016
    Hyderabad, TG, India

    While working for the Linux Performance Team, I designed an Android system app for testing and bench-marking systems.

    Systems Engineering Intern | May 2015 - Jul 2015
    Hyderabad, TG, India

    As a part of the Linux Kernel Team, I implemented a memory crawler app for Linux on python.

    Simulated a Smart Traffic Grid on Sumo (coded in python).

  • IIT Kharagpur

    Teaching Assistant | Jan 2016 - Apr 2017
    Kharagpur, WB, India

    I instructed three undergraduate courses offered by Electrical Engineering department in three consecutive semesters.

    Research Assistant | Jul 2015 - Apr 2017
    Kharagpur, WB, India

    [SEAL] As a research student working in the Secured Embedded Architecture Laboratory in the department of Computer Science, I investigated software acceleration of novel reversible watermarking algorithm, followed by an FPGA accelerator architecture design.

EDUCATION

  • U. of Texas, Austin

    Doctor of Philosophy | Aug 2023 - Present
    Austin, TX, USA

    Currently pursuing a PhD in the department of Electrical & Computer Engineering under Dr. Sandeep Chinchali, with a research focus on energy-efficient, robust machine learning algorithms for generative AI edge applications. Current cumulative grade point average is 4.00

  • U. of Southern California

    Master of Science | Aug 2017 - Jun 2020
    Los Angeles, CA, USA

    M.S. in Computer Science fully funded by the Ming Hsieh Department of ECE graduate assistant-ship.

    Master of Science | Jan 2018 - Apr 2020
    Los Angeles, CA, USA

    Research focussed M.S. in Electrical Engineering fully funded by the Ming Hsieh Department of ECE graduate assistant-ship. Completed PhD level coursework and published in top-tier conference.

  • IIT Kharagpur

    Bachelor & Master of Technology | Jul 2012 - Aug 2017
    Kharagpur, WB, India

    Bachelors of Technology in Electrical Engineering. Master of Technology in Instrumentation & Signal Processing. Thesis: Increasing Execution Throughput of Reversible Watermarking Algorithms: A Hardware Implementation on FPGA. Minor: Electronics & Electrical Communications Engineering. Micro-specialization: Embedded Wireless Systems (GSSST). Ranked 2551 (99.47 %-ile nationwide) in IIT JEE. Recieved M.C.M. scholarship during first four years. Opted for a department change (top 2%-ile, 23 out of 1332).

TECHNICAL SKILLS

Python, C, C++, PyTorch
Tensorflow, Shell, HPC(OpenMP), LATEX
Cuda, OpenCL, Matlab
Java

AFFILIATIONS


CURRENT

UT Austin AMD, Inc.

PREVIOUS

Qualcomm, Inc. Amazon Lab126 MTK Research USC IIT Kgp
GOOGLE SCHOLAR CITATIONS

0

GOOGLE SCHOLAR CITATIONS

STACK EXCHANGE OUTREACH

0

STACK EXCHANGE OUTREACH

HONORS & AWARDS

0

HONORS & AWARDS

RESEARCH BLOGS AND PUBLICATIONS

  • Neural Network Training with Approximate Logarithmic Computations image