AMD 數據科學工具包包含以下組件:
hipDF – 一個 GPU 加速的 DataFrame 庫,提供快速且可擴展的表格數據操作、聚合與轉換。hipDF 支援現代數據管道中所需的高性能預處理、特徵工程和 ETL 工作流程,並能以極少或無需代碼更改的方式加速現有的 Pandas 工作流程。
hipGRAPH – 利用 GPU 加速來快速且精確地處理和分析複雜的圖結構與網絡。hipGRAPH 支援多種圖算法,例如中心性、遍歷、相似性、抽樣和標籤,並與 AMD 數據科學生態系統中的 hipDF DataFrame 無縫整合。
hipMM – HIP 記憶體管理庫提供先進的 GPU 記憶體管理工具,包括高效分配、池化和數據移動,以支援組成 AMD 數據科學工具包的各個庫。
hipRAFT – 提供數據科學與機器學習中可重用的 GPU 加速基元基礎層,包括聚類、維度約減和統計運算。hipRAFT 是更高層次數據科學與 AI 應用的計算核心。
hipVS – 一個 GPU 加速的向量搜尋庫,包含多種高性能近似與精確最近鄰與聚類算法。hipVS 與 hipDF 提供的 DataFrame 無縫整合,以支援廣泛的數據科學工作負載。
dask-hip – 一個用於在支援 HIP 的 AMD GPU 系統上部署和管理 Dask 工作者的庫。dask-hip 為在一個或多個 AMD GPU 上擴展 GPU 加速數據科學工作負載增添分散式執行能力。
hip-ucxx – 一個用於 AMD 平台上分散式 GPU 工作流程的通信組件。hip-ucxx 透過為 ROCm 環境提供基於 UCXX 的通信支援,補充 dask-hip,有助於在 AMD 系統上實現分散式與多 GPU 數據科學工作流程。