ATP: In-network Aggregation for Multi-tenant Learning. - ChonLam Lao; Yanfang Le; Kshiteej Mahajan; Yixi Chen; Wenfei Wu; Aditya Akella; Michael M. Swift (NSDI, 2021)

Efficient Data-Plane Memory Scheduling for In-Network Aggregation. - Hao Wang; Yuxuan Qin; ChonLam Lao; Yanfang Le; Wenfei Wu; Kai Chen (CoRR, 2022)

A Generic Service to Provide In-Network Aggregation for Key-Value Streams. - Yongchao He; Wenfei Wu; Yanfang Le; Ming Liu 027; ChonLam Lao (ASPLOS (2), 2023)

Preemptive Switch Memory Usage to Accelerate Training Jobs with Shared In-Network Aggregation. - Hao Wang; Yuxuan Qin; ChonLam Lao; Yanfang Le; Wenfei Wu; Kai Chen (ICNP, 2023)

In-Network Key-Value Cache with Linearizability. - Yuxuan Qin; Weize Gao; ChonLam Lao; Wenfei Wu; Kai Chen (ICPADS, 2023)

THC: Accelerating Distributed Deep Learning Using Tensor Homomorphic Compression. - Minghao Li 003; Ran Ben Basat; Shay Vargaftik; ChonLam Lao; Kevin Xu; Xinran Tang; Michael Mitzenmacher; Minlan Yu (CoRR, 2023)

THC: Accelerating Distributed Deep Learning Using Tensor Homomorphic Compression. - Minghao Li 003; Ran Ben Basat; Shay Vargaftik; ChonLam Lao; Kevin Xu; Michael Mitzenmacher; Minlan Yu (NSDI, 2024)

HawkVision: Low-Latency Modeless Edge AI Serving. - ChonLam Lao; Jiaqi Gao; Ganesh Ananthanarayanan; Aditya Akella; Minlan Yu (CoRR, 2024)

TrainMover: Efficient ML Training Live Migration with No Memory Overhead. - ChonLam Lao; Minlan Yu; Aditya Akella; Jiamin Cao; Yu Guan; Pengcheng Zhang; Zhilong Zheng; Yichi Xu; Ennan Zhai; Dennis Cai; Jiaqi Gao (CoRR, 2024)

eTran: Extensible Kernel Transport with eBPF. - Zhongjie Chen; Qingkai Meng; ChonLam Lao; Yifan Liu; Fengyuan Ren; Minlan Yu; Yang Zhou (NSDI, 2025)

An Extensible Software Transport Layer for GPU Networking. - Yang Zhou; Zhongjie Chen; Ziming Mao; ChonLam Lao; Shuo Yang; Pravein Govindan Kannan; Jiaqi Gao; Yilong Zhao; Yongji Wu; Kaichao You; Fengyuan Ren; Zhiying Xu; et al. (CoRR, 2025)

Towards Easy and Realistic Network Infrastructure Testing for Large-scale Machine Learning. - Jinsun Yoo; ChonLam Lao; Lianjie Cao; Bob Lantz; Minlan Yu; Tushar Krishna; Puneet Sharma (CoRR, 2025)