Aggify: Lifting the Curse of Cursor Loops using Custom Aggregates. - Surabhi Gupta; Sanket Purandare; Karthik Ramachandra 002 (SIGMOD Conference, 2020)

Optimizing Cursor Loops in Relational Databases. - Surabhi Gupta; Sanket Purandare; Karthik Ramachandra 002 (CoRR, 2020)

μ-TWO: 3× Faster Multi-Model Training with Orchestration and Memory Optimization. - Sanket Purandare; Abdul Wasay; Animesh Jain; Stratos Idreos (MLSys, 2023)

TorchTitan: One-stop PyTorch native solution for production ready LLM pre-training. - Wanchao Liang; Tianyu Liu; Less Wright; Will Constable; Andrew Gu; Chien-Chin Huang; Iris Zhang; Wei Feng; Howard Huang; Junjie Wang; Sanket Purandare; Gokul Nadathur; et al. (CoRR, 2024)

SimpleFSDP: Simpler Fully Sharded Data Parallel with torch.compile. - Ruisi Zhang; Tianyu Liu; Will Feng; Andrew Gu; Sanket Purandare; Wanchao Liang; Francisco Massa (CoRR, 2024)

TorchTitan: One-stop PyTorch native solution for production ready LLM pretraining. - Wanchao Liang; Tianyu Liu; Less Wright; Will Constable; Andrew Gu; Chien-Chin Huang; Iris Zhang; Wei Feng; Howard Huang; Junjie Wang; Sanket Purandare; Gokul Nadathur; et al. (ICLR, 2025)

Flash Inference: Near Linear Time Inference for Long Convolution Sequence Models and Beyond. - Costin-Andrei Oncescu; Sanket Purandare; Stratos Idreos; Sham M. Kakade (ICLR, 2025)