Enabling Switch Memory Management for Distributed Training with In-Network Aggregation | IEEE Conference Publication | IEEE Xplore