Apex

PyTorch 扩展工具包，轻松实现混合精度与分布式训练。