RAPIDS.AI深度学习库
该存储库是我们将GPU上的数据帧的RAPIDS加速集成到流行的深度学习框架中的工作的所在地。 这项工作可以分为三个主要部分:
开发了数据加载器和预处理功能,以帮助提供RAPIDS cuDF数据帧与可用的不同深度学习库之间的连接。
通过融合GPU操作来优化优化器。
在比赛中或在现实世界数据集上使用上述每种方法的示例。
每个深度学习库都包含在其自己的子文件夹中,而不同的数据加载器选项和示例包含在其他子文件夹中。 目前,我们的重点是PyTorch,但是我们希望将来会添加其他库。