GPT-2をWebGLでブラウザ実装
🕒
🔗
元記事を見る
カテゴリ
IT・ネットオープンソースプロジェクト「gpt2-webgl」は、GPT-2の小型モデル(117M)の推論処理をWebGL2を用いてブラウザ上で実現する試みです。事前学習済みの重みをPythonスクリプトで取得し、GPUでの推論をJavaScriptで可能にします。トランスフォーマーブロックや注意行列の可視化機能も備え、機械学習の学習やデモ用途に適しています。MITライセンスで提供されています。