PFVM向けのONNXエクスポータの開発 - Preferred Networks Research & Development本記事は、2023年夏季インターンシッププログラムで勤務された開田亮佑さんによる寄稿です。 はじめに 2023年度夏季インターンの開田亮佑です。この度インターンプロジェクトとして、PFVM向けのONNXエクスポータの開発に取り組みました。特に、エクスポートにおける深層学習モデルの計算グラフの取得に要する時間を短縮することと、エクスポータの拡張やデバッグを容易にすることを目的に開発を行いました。その取り組みを紹介いたします。 背景 図1. 社内で典型的な深層学習の推論ワークロード 図1は、社内で典型的な深層学習の推論ワークロードの概要を模式的に表したものです。ワークロードでは、PyTorchなどのフレームワークで構築されたモデルをランタイムやチップ上で実行しますが、その前に、ONNXといった統一的なフォーマットに変換し、最適化する処理がよく行われます。 PFNではPFVMと呼ばれる深層学習の
