maharshi-pandya/megacute

maharshi-pandya/megacute は Python で Transformer の演算グラフを定義し、CuTe DSL ベースの持続実行カーネルで処理するライブラリです。

GitHub の説明: Megakernels entirely in CuTeDSL

スター

★ 7

フォーク

2

言語

Python

ライセンス

Apache-2.0