floneum/kalosm
Instant, controllable, local pre-trained AI models in Rust
解决的问题
Kalosm 为在 Rust 应用中集成预训练 AI 模型提供了一个简化且统一的接口。它消除了管理不同模型架构和硬件加速的复杂性,使开发者能够轻松部署用于文本、音频和图像处理的本地或远程模型。
工作原理
该项目由两个主要组件构成:
- Kalosm:一个支持多种预训练模型(如 Llama、Mistral、Phi、Whisper 和 BERT)的高级接口,适用于不同模态。它包含从各种文件格式中提取上下文、网络爬取以及使用自定义解析引擎进行结构化生成的工具,可将模型输出约束为特定的 Rust 类型或语法。
- Fusor:基于编译器的量化机器学习推理运行时。它使用 e-graph 编译器将操作链融合为针对 CPU 和 WebGPU 加速器优化的内核,实现无需手动编写着色器代码的高效本地推理。
适用人群
希望使用本地或远程模型构建 AI 驱动应用,但无需深入掌握机器学习或硬件特定优化的 Rust 开发者。
主要亮点
- 多模态支持:集成支持 LLM、音频转录(Whisper)和图像分割(SAM)。
- 结构化生成:通过
#[derive(Parse, Schema)]或自定义语法,将模型输出约束为任意 Rust 类型。 - JSON/HTML/XML 支持:对生成复杂数据格式提供快速、结构感知的加速。
- 硬件加速:通过 Fusor 运行时,针对 CPU 和 WebGPU 实现本地推理优化。
- RAG 工具:内置工具用于从 PDF、HTML 等文档中提取、分块和检索上下文。
相关
- 项目
- 项目
- 项目
- 项目
- 项目