tracel-ai/models

解决的问题

本仓库提供了一个使用 Burn 深度学习框架实现的深度学习模型的集中式集合。它使开发者能够无需从头实现，即可访问各种预构建的模型架构，用于不同的 AI 任务。

工作原理

该项目组织了一系列官方和社区贡献的模型。这些实现基于 Burn 框架构建，支持使用 Rust 实现深度学习。涵盖了文本、图像和音频等多种模态。

适用人群

使用 Burn 框架并需要即用型模型架构来完成语言建模、图像分类和语音识别等任务的开发者和研究人员。

主要亮点

• Llama、ALBERT 和 ResNet 等模型的官方实现。
• 包括 Stable Diffusion、Whisper 和 RWKV v7 在内的社区贡献。
• 支持多种模态：文本（大语言模型、编码器）、视觉（分类、目标检测）、音频（语音识别、音乐音符检测）。
• 采用 MIT 和 Apache 2.0 许可证发布。