nv-legate/cupynumeric

NumPy and SciPy on Multi-Node Multi-GPU systems

解决的问题

解决了将 NumPy 工作流扩展到高性能计算环境的困难。用户无需为分布式系统或 GPU 重写代码,即可在大规模硬件集群上运行现有的基于 NumPy 的数据分析和机器学习工作流。

工作原理

cuPyNumeric 在 Legate 框架之上实现了 NumPy API。这使得它能够在单个 CPU、单个 GPU 以及跨多个节点的数千个 GPU 之间分布计算,而无需用户进行重大代码更改。

适用人群

专为从事大规模数据分析、复杂模拟和机器学习的研究人员和开发者设计,他们需要将计算从单台机器扩展到分布式 GPU 集群。

主要亮点

  • 高性能数组计算
  • 实现了无缝迁移的 NumPy API
  • 从单个 CPU 扩展到数千个 GPU
  • 支持跨多个节点的分布式系统

相关

  • 项目
  • 项目
  • 项目
  • 项目
  • 项目