A RoCE network for distributed AI training at scale