Learning to count: a neural network model of the successor function. - Vima Gupta; Sashank Varma (CogSci, 2022)

Neural Network Models of Becoming a Cardinal Principle Knower. - Vima Gupta; Sashank Varma (CoRR, 2023)

Lynx: Enabling Efficient MoE Inference through Dynamic Batch-Aware Expert Selection. - Vima Gupta; Kartik Sinha; Ada Gavrilovska; Anand Padmanabha Iyer (CoRR, 2024)