Inference by Reparameterization in Neural Population Codes. - Rajkumar Vasudeva Raju; Xaq Pitkow (NIPS, 2016)

Schema-learning and rebinding as mechanisms of in-context learning and emergence. - Sivaramakrishnan Swaminathan; Antoine Dedieu; Rajkumar Vasudeva Raju; Murray Shanahan; Miguel Lázaro-Gredilla; Dileep George (NeurIPS, 2023)

Graph schemas as abstractions for transfer learning, inference, and planning. - J. Swaroop Guntupalli; Rajkumar Vasudeva Raju; Shrinu Kushagra; Carter Wendelken; Danny Sawyer; Ishan Deshpande; Guangyao Zhou; Miguel Lázaro-Gredilla; Dileep George (CoRR, 2023)

Inferring Inference. - Rajkumar Vasudeva Raju; Zhe Li 002; Scott Linderman; Xaq Pitkow (CoRR, 2023)

Diffusion Model Predictive Control. - Guangyao Zhou; Sivaramakrishnan Swaminathan; Rajkumar Vasudeva Raju; J. Swaroop Guntupalli; Wolfgang Lehrach; Joseph Ortiz; Antoine Dedieu; Miguel Lázaro-Gredilla; Kevin Patrick Murphy (Trans. Mach. Learn. Res., 2025)