Shreyas Grampurohit

B.Tech. & M.Tech. in Electrical Engineering

Indian Institute of Technology Bombay

Shreyas Grampurohit

About Me

I am a final-year Dual Degree (B.Tech. + M.Tech.) student in the Department of Electrical Engineering at IIT Bombay, specializing in Communication and Signal Processing with a minor in Computer Science and Engineering.

My research interests lie at the intersection of deep learning and signal processing, including learning-based inverse problems, compressive sensing, and interpretable recovery, as well as ML for systems and computer architecture.

I have had the privilege of collaborating with Prof. Ajit Rajwade and Prof. Satish Mulleti at IIT Bombay, and Prof. Vijay Janapa Reddi at the Edge Computing Lab, Harvard University.

Education

IIT Bombay
B.Tech. & M.Tech. in EE
Minor in CS&E
CPI: 8.99/10
Nov 2021 — Jun 2026

Honors

Best of CAL at HPCA 2026
Best Poster at RISC 2025
KVPY Fellow (AIR 198)
JEE Advanced AIR 495

Interests

Inverse Problems
Compressive Sensing
Interpretable Recovery
ML for Architecture

Recent Publications

QuArch: Question-Answering Dataset for AI Agents in Computer Architecture

IEEE Computer Architecture Letters (CAL), 2025

Best of CAL at HPCA 2026

Two-Dimensional Tomographic Reconstruction From Projections

IEEE ICASSP 2026

Verifiable Deep Quantitative Group Testing

Preprint, 2025

Research Experience

Deep Learning for Group Testing

Apr 2025 — Present

Dual Degree Thesis | Guide: Prof. Ajit Rajwade & Prof. Satish Mulleti

  • Designed a noise-robust Deep Learning solver for Quantitative Group Testing, achieving reconstruction performance matching state-of-the-art combinatorial algorithms
  • Established structural verifiability by introducing novel Jacobian analysis, showcasing that the network implicitly internalizes the combinatorial structure of the problem
  • Demonstrated superior signal recovery resilience in regimes with sparse measurement noise compared to model-based solvers like Approximate Message Passing
  • Extending the framework to real-valued signal recovery for applications like viral load estimation with rigorous theoretical guarantees

Archipedia: Generative AI for Hardware Design

Jan 2024 — Present

In-semester Research | Guide: Prof. Vijay Janapa Reddi, Harvard University

  • Spearheaded "Archipedia," a one-billion-token dataset covering 50 years of premier computer architecture research, leading a team of 10 students
  • Co-authored QuArch benchmark for evaluating LLMs in computer architecture; paper accepted to IEEE CAL and selected for 'Best of CAL' at HPCA 2026
  • Architected metadata extraction and validation pipelines on AWS to parse thousands of academic manuscripts

Tomography Under Unknown Angles and Spatial Shifts

Aug 2024 — Mar 2025

Research Project | Guide: Prof. Ajit Rajwade & Prof. Satish Mulleti

  • Developed a robust reconstruction framework for Cryo-EM Tomography addressing the ill-posed inverse problem of recovering 2D images from corrupted projections
  • Formulated a three-way alternating minimization algorithm with shift-aware modified Graph Laplacian initialization
  • Proved theoretical non-feasibility of spectral invariant estimators for arbitrary translational shifts
  • Achieved 2.36× lower RRMSE compared to baselines; Best Poster Award at RISC 2025

ArchGym: ML-Assisted Architecture Design

Jun 2023 — Aug 2023

Research Internship | Guide: Prof. Vijay Janapa Reddi, Harvard University

  • Contributed to ArchGym, an open-source framework connecting diverse ML algorithms to architecture simulators
  • Developed ML-based proxy models to mitigate simulator latency, optimizing the accuracy-speed trade-off
  • Led automated documentation generation and authored comprehensive guides for research reproducibility

Professional Experience

Digital Electronics Intern — Texas Instruments

May 2024 — Jul 2024

Radar Business Unit | Bengaluru, India

  • Conducted comprehensive performance benchmarking of radar chip architectures using Coremark and Dhrystone suites
  • Analyzed cache utilization metrics on a QuickTurn emulator, implementing modifications to cache interactions
  • Extracted key performance metrics enabling projected 30% reduction in power and 40% lower area utilization

Academic Projects

Neutrophil Image Generation with Human Feedback

ML in Healthcare | Prof. Kshitij Jadhav

Trained diffusion model for synthetic histopathology images. Designed multi-label reward model from expert pathologist feedback, improving plausibility from 82% to 90%.

Jan — May 2024

Embedded DSP System Design

DSP Lab | Prof. Preeti Rao

Optimized real-time FIR & IIR filters on TI ezDSP5515. Accelerated spectrum analysis with DMA ping-pong buffers and developed DTMF decoder with GNU Radio.

Jan — Apr 2025

Reinforcement Learning & MDPs

FILA | Prof. Shivaram Kalyanakrishnan

Implemented UCB, KL-UCB, Thompson Sampling on bandits. Built MDP planner and designed RL agent with CMA-ES for autonomous driving.

Jan — Apr 2025

Denoising Diffusion Probabilistic Models

Advanced ML | Prof. Amit Sethi

Implemented DDPM with custom U-Net architecture. Explored noise scheduling strategies and extended to color image generation in HSV space.

Aug — Nov 2023

Energy-efficient Spatial Data Prefetcher

Advanced Architecture | Prof. Biswabandan Panda

Improved energy-efficiency of Bingo Spatial Data Prefetcher achieving 90% accuracy while maintaining speedup. Evaluated using ChampSim simulator.

Aug — Nov 2024

Parallel Retinal Neuron Network Simulation

HPC | Prof. Shivasubramanian Gopalakrishnan

Accelerated neuron network simulations with OpenMP, achieving 4.6× speedup. Scaled to 1000+ computational units with detailed profiling.

Aug — Nov 2024

FPGA-based RRAM Characterization

Electronic Design Lab | Prof. Siddharth Tallur

Led analog frontend design for FPGA-based RRAM system. Developed Python GUI for real-time control and data acquisition, achieving <5% error.

Jan — May 2024

Stereo Vision & Shape from Shading

Computer Vision | Prof. Subhasis Chaudhuri

Implemented stereo vision algorithms on Middlebury dataset. Developed shape-from-shading for 3D geometry reconstruction under varying conditions.

Jan — Apr 2025

Teaching & Mentorship

Teaching Assistant — Signal Processing

Prof. Preeti Rao, Dept. of Electrical Engineering, IIT Bombay

100+ students | Aug — Nov 2025

Summer of Science Mentor — Image Processing

Maths and Physics Club, IIT Bombay

Designed curriculum and guided 7 students | May — Jul 2025

Technical Skills

Languages & Tools

Python C C++ MATLAB VHDL Assembly GNURadio Quartus Prime Keil µVision

Libraries

PyTorch NumPy Scikit-Learn Pandas Matplotlib SciPy OpenCV Selenium OpenAI Gym

Infrastructure & HPC

OpenMP MPI CUDA C LaTeX NgSPICE Google Cloud Platform AWS

Try searching for "tomography", "computer architecture", "pytorch", or any keyword