Learning Robust Dynamics through Variational Sparse Gating. - Arnav Kumar Jain; Shivakanth Sujit; Shruti Joshi; Vincent Michalski; Danijar Hafner; Samira Ebrahimi Kahou (NeurIPS, 2022)

Factorized multi-scale multi-resolution residual network for single image deraining. - Shivakanth Sujit; Deivalakshmi Subbian; Seok-Bum Ko (Appl. Intell., 2022)

Reinforcement learning based adaptive PID controller design for control of linear/nonlinear unstable processes. - T. Shuprajhaa; Shivakanth Sujit; K. Srinivasan 001 (Appl. Soft Comput., 2022)

RelationalUNet for Image Segmentation. - Ivaxi Sheth; Pedro H. M. Braga; Shivakanth Sujit; Sahar Dastani; Samira Ebrahimi Kahou (MLMI@MICCAI (1), 2023)

Prioritizing Samples in Reinforcement Learning with Reducible Loss. - Shivakanth Sujit; Somjit Nath; Pedro H. M. Braga; Samira Ebrahimi Kahou (NeurIPS, 2023)

Bridging the Gap Between Offline and Online Reinforcement Learning Evaluation Methodologies. - Shivakanth Sujit; Pedro H. M. Braga; Jörg Bornschein; Samira Ebrahimi Kahou (Trans. Mach. Learn. Res., 2023)

PcLast: Discovering Plannable Continuous Latent States. - Anurag Koul; Shivakanth Sujit; Shaoru Chen; Ben Evans; Lili Wu; Byron Xu; Rajan Chari; Riashat Islam; Raihan Seraj; Yonathan Efroni; Lekan P. Molu; Miroslav Dudík; et al. (ICML, 2024)

Reinforcement Learning for Sequence Design Leveraging Protein Language Models. - Jithendaraa Subramanian; Shivakanth Sujit; Niloy Irtisam; Umong Sain; Derek Nowrouzezahrai; Samira Ebrahimi Kahou; Riashat Islam (CoRR, 2024)

Improving Low-Cost Teleoperation: Augmenting GELLO with Force. - Shivakanth Sujit; Luca Nunziante; Dan Ogawa Lillrank; Rousslan Fernand Julien Dossa; Kai Arulkumaran (SII, 2025)