您即将离开知乎，请注意您的账号和财产安全。
https://github.aichem.org/tczhangzhi/pytorch-distributed/blob/master/apex_distributed.py
