Kernelllm GGUF
KernelLLM是基於Llama 3.1 Instruct微調的模型,專門用於編寫使用Triton的GPU內核。
下載量 214
發布時間 : 5/19/2025
模型概述
該模型能夠將PyTorch模塊轉換為Triton內核,並在KernelBench-Triton數據集上進行了評估,表現出競爭性或更優的性能。
模型特點
128k tokens上下文長度
支持超長上下文,適合處理複雜的內核生成任務
GPU內核代碼生成
專門優化用於將PyTorch模塊轉換為高效的Triton GPU內核
高效性能
相比更大的模型,在內核生成任務上表現出競爭性或更優的性能
模型能力
文本生成
代碼生成
GPU內核優化
使用案例
GPU編程
PyTorch到Triton轉換
將PyTorch模塊自動轉換為高效的Triton GPU內核
生成的內核在性能上可與手工優化代碼媲美
高性能計算
為科學計算和機器學習任務生成優化的GPU內核
提升計算密集型任務的執行效率
精選推薦AI模型
Llama 3 Typhoon V1.5x 8b Instruct
專為泰語設計的80億參數指令模型,性能媲美GPT-3.5-turbo,優化了應用場景、檢索增強生成、受限生成和推理任務
大型語言模型
Transformers 支持多種語言

L
scb10x
3,269
16
Cadet Tiny
Openrail
Cadet-Tiny是一個基於SODA數據集訓練的超小型對話模型,專為邊緣設備推理設計,體積僅為Cosmo-3B模型的2%左右。
對話系統
Transformers 英語

C
ToddGoldfarb
2,691
6
Roberta Base Chinese Extractive Qa
基於RoBERTa架構的中文抽取式問答模型,適用於從給定文本中提取答案的任務。
問答系統 中文
R
uer
2,694
98