NeSSA: Near-Storage Data Selection for Accelerated Machine Learning Training. - Neha Prakriya; Yu Yang 007; Baharan Mirzasoleiman; Cho-Jui Hsieh; Jason Cong (HotStorage, 2023)

TAPA-CS: Enabling Scalable Accelerator Design on Distributed HBM-FPGAs. - Neha Prakriya; Yuze Chi; Suhail Basalama; Linghao Song; Jason Cong (ASPLOS (3), 2024)

SpectraFlux: Harnessing the Flow of Multi-FPGA in Mass Spectrometry Clustering. - Tianqi Zhang; Neha Prakriya; Sumukh Pinge; Jason Cong; Tajana Rosing (DAC, 2024)

HMT: Hierarchical Memory Transformer for Long Context Language Processing. - Zifan He; Zongyue Qin; Neha Prakriya; Yizhou Sun; Jason Cong (CoRR, 2024)

Multi-Token Joint Speculative Decoding for Accelerating Large Language Model Inference. - Zongyue Qin; Ziniu Hu; Zifan He; Neha Prakriya; Jason Cong; Yizhou Sun (CoRR, 2024)

Accelerating Large Language Model Pretraining via LFR Pedagogy: Learn, Focus, and Review. - Neha Prakriya; Jui-Nan Yen; Cho-Jui Hsieh; Jason Cong (CoRR, 2024)

Dynamic-Width Speculative Beam Decoding for Efficient LLM Inference. - Zongyue Qin; Zifan He; Neha Prakriya; Jason Cong; Yizhou Sun (CoRR, 2024)

Dynamic-Width Speculative Beam Decoding for LLM Inference. - Zongyue Qin; Zifan He; Neha Prakriya; Jason Cong; Yizhou Sun (AAAI, 2025)

Optimized Multi-Token Joint Decoding With Auxiliary Model for LLM Inference. - Zongyue Qin; Ziniu Hu; Zifan He; Neha Prakriya; Jason Cong; Yizhou Sun (ICLR, 2025)

HMT: Hierarchical Memory Transformer for Efficient Long Context Language Processing. - Zifan He; Yingqi Cao; Zongyue Qin; Neha Prakriya; Yizhou Sun; Jason Cong (NAACL (Long Papers), 2025)

LIFT: LLM-Based Pragma Insertion for HLS via GNN Supervised Fine-Tuning. - Neha Prakriya; Zijian Ding; Yizhou Sun; Jason Cong (CoRR, 2025)