メインコンテンツへスキップ
buildradar
ログイン

tspeterkim/flash-attention-minimal

@tspeterkim

約100行の CUDA で実装された Flash Attention(フォワードパスのみ)

スター
1,181
フォーク
117
言語
Cuda
ライセンス
Apache-2.0
最終プッシュ
2 年前
Cuda

関連 Intel はまだありません

このリポジトリは radar が追跡するソースにまだ登場していません。コレクターはスケジュールで動いています——このリポジトリがカバーされたらまた見てください。