huggingface/kernels

Build compute kernels and load them from the Hub.

解決的問題

它為 Python 套件與應用程式提供了一種直接從 Hugging Face Hub 動態載入最佳化計算內核的方法,無需透過 PYTHONPATH 進行傳統的 Python 套件安裝。

作用方式

此專案包含一個名為 kernels 的 Python 套件,讓使用者可從 Hub 下載並執行內核。對於開發者,還包含一個 kernel-builder 工具,用於建立這些可移植且具版本控制的內核。這些內核設計為可在不同 Python 版本、PyTorch 編譯設定(包含各種 CUDA 版本與 C++ ABIs)以及較舊的 C 套件版本之間跨平台運作。

適用對象

  • AI 開發者:需要為應用程式提供高效率計算內核,但又不願管理複雜安裝流程的人。
  • 內核作者:希望建立並分發可在各種環境中相容的最佳化計算內核的開發者。

主要特色

  • 動態載入:直接從 Hugging Face Hub 載入內核。
  • 可移植性:可從標準 Python 路徑以外的路徑載入內核。
  • 版本控制:支援在單一 Python 進程中載入同一內核的多個版本。
  • 相容性:廣泛支援最新 Python 版本以及各種 PyTorch/CUDA 設定。

相關

  • Dispatch
  • Dispatch
  • Dispatch
  • 專案
  • Dispatch