pytorch/kineto

何を解決するか

Kineto は、機械学習ワークロードのパフォーマンス可視化と診断を提供し、開発者が一般的な ML コンポーネントにおけるボトルネックを特定・解決するのを支援します。

動作方法

プロファイリングライブラリとして機能し、そのコアコンポーネントである Libkineto は、低オーバーヘッドの GPU タイムライントレーシングに焦点を当てています。PyTorch Profiler に直接統合されており、システムレベルのプロファイリングとパフォーマンス問題に対する実行可能な推奨事項を提供します。

対象ユーザー

PyTorch を使用して機械学習モデルを開発・最適化する ML エンジニアや開発者で、GPU のパフォーマンスを分析し、システムレベルのボトルネックを診断したい方。

主な特徴

• Libkineto を通じた低オーバーヘッドの GPU タイムライントレーシング。
• 外部のシステムレベルプロファイリングツールとの統合。
• 人気の可視化プラットフォームや分析パイプラインとの互換性。
• 一般的なパフォーマンス問題に対する実行可能な推奨事項を提供。