概要
Hugging Faceが、カスタムカーネル(GPUなどの計算処理を高速化する専用コード)の配布・利用を標準化する「🤗 Kernels」プロジェクトの大型アップデートを発表しました。セキュリティ強化や新しいリポジトリタイプの追加が主な内容です。
何が発表・更新されたのか
Hub上に「kernel」という新しいリポジトリタイプが追加され、対応アクセラレータやOS、バックアージョンが一覧で確認できるようになりました。あわせて、信頼できる公開者(trusted publishers)やコード署名によるセキュリティ強化、CLIの再設計、対応フレームワークの拡充などが行われています。
なぜ重要なのか
カーネルはPythonプロセスと同じ権限でネイティブコードを実行するため、悪意あるコードが実害を及ぼす可能性があります。今回、既定では信頼できる公開者のカーネルのみを読み込む仕組みや、Sigstoreのcosignを使った署名が導入され、安全性が高まりました。
誰に関係があるのか
主にGPUカーネルを開発・利用する機械学習エンジニアや研究者が対象です。高速化された処理を安全に配布・再利用したい開発者に関係します。
仕事や業務でどう使えるのか
Hub上のカーネル一覧から用途に合ったカーネルを探し、get_kernelで読み込んで利用できます。エージェント(自律的にタスクを進めるAI)によるカーネルの自動生成・ベンチマークを支える基盤づくりも進められています。
注意点
信頼できない公開者のカーネルを読み込むにはtrust_remote_code=Trueの明示指定が必要です。またコード署名はkernels verify-signatureで検証できますが、読み込み時の自動署名検証はまだ本格導入前とされています。
公式情報
関連キーワード
- Hugging Face
- GPUカーネル
- PyTorch(Torch Stable ABI)
- コード署名(Sigstore / cosign)
- AIエージェント

