singa

收录理由

Apache SINGA 是一个面向分布式深度学习的开源库，专为希望跨多块 GPU 训练模型、又不想自行搭建集群基础设施的团队设计。它支持单机或多节点上的数据并行训练，内置计算图机制，能够自动完成反向传播，并提供了 Adam、RMSProp 以及带动量的 SGD 等常用优化器。它的两个特点值得关注：一是通过 ONNX 交换模型，便于与其他工具链互操作；二是可以将训练好的模型作为存储过程嵌入关系数据库，直接在应用数据所在的位置执行预测。这种设计让研究团队和工程团队能够轻松地让训练结果紧邻业务数据被查询和使用。