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は、1つ以上のAMD GPUにまたがってGPUアクセラレーションされたデータサイエンスワークロードをスケーリングするための分散実行機能を追加します。
hip-ucxx – AMDプラットフォーム上で分散GPUワークフローを実現するための通信コンポーネントです。hip-ucxxは、ROCm環境向けにUCXXベースの通信サポートを提供することで、dask-hipを補完し、AMDシステム全体での分散およびマルチGPUデータサイエンスワークフローの実現を支援します。