tracel-ai/models
Models and examples built with Burn
解決的問題
本儲存庫提供了一個使用 Burn 深度學習框架實作的深度學習模型集中式集合。讓開發者無需從頭開始實作,即可存取各種預先建構的模型架構,用於不同的 AI 任務。
運作方式
此專案整理了一系列官方與社群貢獻的模型。這些實作皆建構於 Burn 框架之上,支援使用 Rust 進行深度學習。涵蓋了文本、影像與音訊等多種模態。
適用對象
使用 Burn 框架,並需要即時可用的模型架構來執行語言建模、影像分類與語音辨識等任務的開發者與研究人員。
主要亮點
- Llama、ALBERT 與 ResNet 等模型的官方實作。
- 社群貢獻包括 Stable Diffusion、Whisper 與 RWKV v7。
- 支援多種模態:文字(大語言模型、編碼器)、視覺(分類、物件偵測)、音訊(語音辨識、音樂音符偵測)。
- 以 MIT 與 Apache 2.0 許可證發行。
相關
- 專案
- 專案
- 專案
- 專案
- 專案