Extending Sparse Tensor Accelerators to Support Multiple Compression Formats. - Eric Qin 001; Geonhwa Jeong; William Won; Sheng-Chun Kao; Hyoukjun Kwon; Sudarshan Srinivasan; Dipankar Das 002; Gordon Euhyun Moon; Sivasankaran Rajamanickam; Tushar Krishna (IPDPS, 2021)

Exploring Multi-dimensional Hierarchical Network Topologies for Efficient Distributed Training of Trillion Parameter DL Models. - William Won; Saeed Rashidi; Sudarshan Srinivasan; Tushar Krishna (CoRR, 2021)

Themis: a network bandwidth-aware collective scheduling policy for distributed training of DL models. - Saeed Rashidi; William Won; Sudarshan Srinivasan; Srinivas Sridharan 002; Tushar Krishna (ISCA, 2022)

ASTRA-sim2.0: Modeling Hierarchical Networks and Disaggregated Systems for Large-model Training at Scale. - William Won; Taekyung Heo; Saeed Rashidi; Srinivas Sridharan 002; Sudarshan Srinivasan; Tushar Krishna (ISPASS, 2023)

TACOS: Topology-Aware Collective Algorithm Synthesizer for Distributed Training. - William Won; Midhilesh Elavazhagan; Sudarshan Srinivasan; Ajaya Durg; Swati Gupta 001; Tushar Krishna (CoRR, 2023)

Towards a Standardized Representation for Deep Learning Collective Algorithms. - Jinsun Yoo; William Won; Meghan Cowan; Nan Jiang; Benjamin Klenk; Srinivas Sridharan 002; Tushar Krishna (HOTI, 2024)

LIBRA: Enabling Workload-Aware Multi-Dimensional Network Topology Optimization for Distributed Training of Large AI Models. - William Won; Saeed Rashidi; Sudarshan Srinivasan; Tushar Krishna (ISPASS, 2024)

TACOS: Topology-Aware Collective Algorithm Synthesizer for Distributed Machine Learning. - William Won; Midhilesh Elavazhagan; Sudarshan Srinivasan; Swati Gupta 001; Tushar Krishna (MICRO, 2024)

FRED: Flexible REduction-Distribution Interconnect and Communication Implementation for Wafer-Scale Distributed Training of DNN Models. - Saeed Rashidi; William Won; Sudarshan Srinivasan; Puneet Gupta 001; Tushar Krishna (CoRR, 2024)

FRED: A Wafer-scale Fabric for 3D Parallel DNN Training. - Saeed Rashidi; William Won; Sudarshan Srinivasan; Puneet Gupta 001; Tushar Krishna (ISCA, 2025)

Toward a Standardized Representation for Deep Learning Collective Algorithms. - Jinsun Yoo; William Won; Meghan Cowan; Nan Jiang; Benjamin Klenk; Srinivas Sridharan 002; Tushar Krishna (IEEE Micro, 2025)

MCMComm: Hardware-Software Co-Optimization for End-to-End Communication in Multi-Chip-Modules. - Ritik Raj; Shengjie Lin; William Won; Tushar Krishna (CoRR, 2025)

COSMIC: Enabling Full-Stack Co-Design and Optimization of Distributed Machine Learning Systems. - Aditi Raju; Jared Ni; William Won; Changhai Man; Srivatsan Krishnan; Srinivas Sridharan 002; Amir Yazdanbakhsh; Tushar Krishna; Vijay Janapa Reddi (CoRR, 2025)