QuAFL: Federated Averaging Can Be Both Asynchronous and Communication-Efficient. - Hossein Zakerinia; Shayan Talaei; Giorgi Nadiradze; Dan Alistarh (CoRR, 2022)

Hybrid Decentralized Optimization: First- and Zeroth-Order Optimizers Can Be Jointly Leveraged For Faster Convergence. - Shayan Talaei; Giorgi Nadiradze; Dan Alistarh (CoRR, 2022)

Communication-Efficient Federated Learning With Data and Client Heterogeneity. - Hossein Zakerinia; Shayan Talaei; Giorgi Nadiradze; Dan Alistarh (AISTATS, 2024)

CHESS: Contextual Harnessing for Efficient SQL Synthesis. - Shayan Talaei; Mohammadreza Pourreza; Yu-Chen Chang; Azalia Mirhoseini; Amin Saberi (CoRR, 2024)

OptiMUS-0.3: Using Large Language Models to Model and Solve Optimization Problems at Scale. - Ali AhmadiTeshnizi; Wenzhi Gao; Herman Brunborg; Shayan Talaei; Madeleine Udell (CoRR, 2024)

Hybrid Decentralized Optimization: Leveraging Both First- and Zeroth-Order Optimizers for Faster Convergence. - Shayan Talaei; Matin Ansaripour; Giorgi Nadiradze; Dan Alistarh (AAAI, 2025)

CHASE-SQL: Multi-Path Reasoning and Preference Optimized Candidate Selection in Text-to-SQL. - Mohammadreza Pourreza; Hailong Li; Ruoxi Sun 002; Yeounoh Chung; Shayan Talaei; Gaurav Tarlok Kakkar; Yu Gan; Amin Saberi; Fatma Ozcan; Sercan Ö. Arik (ICLR, 2025)

Reasoning-SQL: Reinforcement Learning with SQL Tailored Partial Rewards for Reasoning-Enhanced Text-to-SQL. - Mohammadreza Pourreza; Shayan Talaei; Ruoxi Sun 002; Xingchen Wan; Hailong Li; Azalia Mirhoseini; Amin Saberi; Sercan Ö. Arik (CoRR, 2025)

SPRINT: Enabling Interleaved Planning and Parallelized Execution in Reasoning Models. - Emil Biju; Shayan Talaei; Zhemin Huang; Mohammadreza Pourreza; Azalia Mirhoseini; Amin Saberi (CoRR, 2025)

When to Trust Context: Self-Reflective Debates for Context Reliability. - Zeqi Zhou; Fang Wu; Shayan Talaei; Haokai Zhao; Cheng Meixin; Tinson Xu; Amin Saberi; Yejin Choi (CoRR, 2025)

StorySage: Conversational Autobiography Writing Powered by a Multi-Agent Framework. - Shayan Talaei; Meijin Li; Kanu Grover; James Kent Hippler; Diyi Yang; Amin Saberi (CoRR, 2025)