Awesome-Rust-MachineLearning

GitHub
2.2k 123 较难 1 次阅读 今天MIT开发框架语言模型图像
AI 解读 由 AI 自动生成,仅供参考

Awesome-Rust-MachineLearning 是一个专为 Rust 语言打造的机器学习资源宝库。它系统性地整理了用 Rust 编写的各类机器学习库、算法实现,并涵盖了相关的博客、书籍、论文及技术讨论。

在 Python 主导的机器学习领域,许多开发者渴望利用 Rust 的高性能与内存安全特性来优化生产环境,却往往面临生态分散、优质库难寻的痛点。Awesome-Rust-MachineLearning 正是为了解决这一难题而生,它充当了从 Python 迁移至 Rust 的桥梁,帮助用户快速定位成熟可靠的工具链。

这份清单特别适合正在探索 Rust 在 AI 领域应用的开发者、研究人员以及系统架构师。无论是需要基础的数据处理(如 DataFrame、向量运算),还是进阶的深度学习、自然语言处理、强化学习及自动机器学习(AutoML)方案,这里都提供了详尽的分类索引。

其独特亮点在于不仅收录了活跃维护的主流项目,也包含了一些虽已停止更新但具有参考价值的代码库,并对每个类别中的优秀库进行了特别标注和代码点评。此外,它还覆盖了 Jupyter Notebook 支持、GPU 加速及可视化绘图等配套工具,为用户构建完整的 Rust 机器学习工作流提供了全方位指引,让开发者能更高效地发现并利用 Rust 在机器学习领域的最佳实践。

使用场景

某金融科技团队正试图将核心风控模型从 Python 迁移至 Rust,以利用其内存安全特性降低生产环境延迟并减少运维成本。

没有 Awesome-Rust-MachineLearning 时

  • 选型迷茫:开发者在 GitHub 上盲目搜索,难以区分哪些机器学习库是活跃维护的,哪些已被废弃,极易踩坑。
  • 生态割裂:缺乏统一指引,找不到与 Rust 原生数据结构(如 DataFrame、Vector)完美配套的预处理和绘图工具,导致数据流水线断裂。
  • 迁移成本高:从 Python 转来的工程师不熟悉 Rust 生态,无法快速找到类似 Scikit-learn 的综合库或对应的最佳实践代码注释。
  • 性能优化困难:不清楚哪些库支持 GPU 加速或针对特定算法(如梯度提升、NLP)有高性能实现,只能重复造轮子。

使用 Awesome-Rust-MachineLearning 后

  • 精准选型:直接查阅分类清晰的清单,快速锁定如 linfa 等高质量、持续维护的核心库,避开过时项目。
  • 全链路打通:依据"Support Tools"章节,迅速整合 Jupyter 内核、数据处理及可视化组件,构建完整的 Rust 机器学习工作流。
  • 平滑过渡:参考针对 Python 迁移者的特别标注和代码评论,团队能迅速理解库的用法,大幅缩短学习曲线。
  • 性能最大化:通过"GPU"和"Deep Neural Network"等分类,直接采用经过验证的高性能方案,确保风控系统低延迟运行。

Awesome-Rust-MachineLearning 充当了 Rust 机器学习生态的“导航图”,让团队从混乱的探索者变为高效的构建者,显著加速了高性能 AI 系统的落地。

运行环境要求

GPU

未说明

内存

未说明

依赖
notes这是一个 Rust 机器学习库的精选列表(Awesome List),而非单一的可执行软件或框架。因此没有统一的运行环境需求。用户需根据列表中具体选择的某个库(如 polars, tch-rs, burn 等)查阅其各自的文档以获取具体的系统、GPU 及依赖要求。主要开发环境需安装 Rust 工具链。
python不适用 (该项目为 Rust 库集合,非 Python 项目)
Rust 工具链
evcxr (可选,用于 Jupyter)
ndarray 或 nalgebra (线性代数)
polars (数据处理)
image-rs (图像处理)
Awesome-Rust-MachineLearning hero image

快速开始

arml

这个仓库是一个用 Rust 编写的机器学习库列表。 它汇集了 GitHub 仓库、博客、书籍、电影、讨论和论文等内容。 该仓库面向那些考虑从 Python 迁移到 Rust 的开发者。🦀🐍

仓库按基础库和算法类别进行了划分。 此外,还包含一些已停止维护的库以及小型库。 对代码中实用的部分进行了注释说明, 并对每个类别中的优秀库进行了点评。

通过这些资源,我们可以找到更优的 Rust 在机器学习领域的使用方式。

目录

支持工具

Jupyter Notebook

evcxr 可以作为 Jupyter 内核或 REPL 使用,非常有助于学习和验证。

图表绘制

目前可以尝试使用 plotters

ASCII 线形图:

示例:

向量

大多数情况下会使用 ndarraystd::vec

此外,也可以关注 nalgebra。当矩阵的大小已知时,它是一个不错的选择。 参见:ndarray 与 nalgebra 的比较 - Reddit

数据框

目前可以尝试使用 polarsdatafusion 也是一个不错的选择。

图像处理

目前可以尝试使用 image-rs。其他库中也实现了诸如线性变换之类的算法。

自然语言处理(预处理)

  • google-research/deduplicate-text-datasets - 该仓库包含用于去重语言模型数据集的代码,如凯瑟琳·李、达芙妮·伊波利托、安德鲁·尼斯特罗姆、张驰远、道格拉斯·埃克、克里斯·卡利森-伯奇和尼古拉斯·卡林尼在论文《Deduplicating Training Data Makes Language Models Better》中所述。此仓库既包含了用 Rust 编写的 ExactSubstr 去重实现,也包含了我们在论文中用来执行去重并检查结果的 Python 脚本。在即将发布的更新中,我们将添加文件以复现 C4、RealNews、LM1B 和 Wiki-40B-en 数据集的 NearDup 去重版本。
  • pemistahl/lingua-rs - 👄 Rust 生态系统中最准确的自然语言检测库,适用于长短文本。
  • usamec/cntk-rs - Microsoft CNTK 库的封装。
  • stickeritis/sticker - 一种基于 LSTM/Transformer/扩张卷积的序列标注器。
  • tensordot/syntaxdot - 神经语法标注器,支持序列标注、词形还原和依存句法分析。
  • christophertrml/rs-natural - Rust 的自然语言处理工具。
  • bminixhofer/nnsplit - 语义文本分割。用于句子边界检测、复合句拆分等。
  • greyblake/whatlang-rs - Rust 的自然语言检测库。
  • finalfusion/finalfrontier - 基于子词的上下文敏感词嵌入。使用 Rust 实现。
  • bminixhofer/nlprule - 一个用 Rust 编写的快速、低资源消耗的自然语言处理与错误纠正库。
  • rth/vtext - 使用 Python 绑定的简单 Rust NLP 工具。
  • tamuhey/tokenizations - 强健且快速的 Rust 和 Python 令牌对齐库。
  • vgel/treebender - 一款受 HDPSG 启发、用 Rust 编写的符号化自然语言解析器。
  • reinfer/blingfire-rs - BlingFire 分词库的 Rust 封装。
  • CurrySoftware/rust-stemmers - 多种语言中的常用停用词。
  • cmccomb/rust-stop-words - 多种语言中的常用停用词。
  • Freyskeyd/nlp - Rust-nlp 是一个使用 Rust 进行自然语言处理算法的库。
  • Daniel-Liu-c0deb0t/uwu - 西方最快的文本 uwu 化工具。

图模型

接口、流水线与 AutoML

工作流

GPU

综合类(类似 scikit-learn)

所有库都支持以下算法。

  • 线性回归
  • 逻辑回归
  • K 均值聚类
  • 神经网络
  • 高斯过程回归
  • 支持向量机
  • 高斯混合模型
  • 朴素贝叶斯分类器
  • DBSCAN
  • k 近邻分类器
  • 主成分分析
  • 决策树
  • 支持向量机
  • 朴素贝叶斯
  • 弹性网络

目前可以尝试使用 smartcorelinfa

综合类(统计学)

梯度提升

深度神经网络

最常用的有 TensorFlow 绑定PyTorch 绑定tch-rs 还提供了 torch vision,非常实用。

图模型

  • Synerise/cleora - Cleora AI 是一种通用模型,用于高效、可扩展地学习异构关系数据中的稳定且归纳性的实体嵌入。
  • Pardoxa/net_ensembles - 用于随机图集合的 Rust 库

自然语言处理(模型)

推荐系统

信息检索

全文检索

最近邻搜索

  • Enet4/faiss-rs - Faiss 的 Rust 语言绑定。
  • rust-cv/hnsw - HNSW 近似最近邻算法,源自论文“利用分层可导航小世界图高效稳健地进行近似最近邻搜索”。
  • hora-search/hora - 🚀 用 Rust 🦀 实现的高效近似最近邻搜索算法库。horasearch.com。
  • InstantDomain/instant-distance - 基于 HNSW 索引的 Rust 高速近似最近邻搜索。
  • lerouxrgd/ngt-rs - NGT 近似最近邻搜索的 Rust 封装。
  • granne/granne - 基于图的近似最近邻搜索。
  • u1roh/kd-tree - Rust 实现的 k 维树。速度快、简单易用。
  • qdrant/qdrant - Qdrant 是一款支持扩展过滤功能的向量相似度搜索引擎。
  • rust-cv/hwt - 来自论文“在汉明空间中进行在线最近邻搜索”的汉明权重树。
  • fulara/kdtree-rust - Rust 版本的 k-d 树实现。
  • mrhooray/kdtree-rs - 用于快速地理空间索引和查询的 Rust k 维树。
  • kornelski/vpsearch - 用于在集合中查找最近(最相似)元素的 C 库。
  • petabi/petal-neighbors - 包括球树和视点树在内的最近邻搜索算法。
  • ritchie46/lsh-rs - Rust 实现的局部敏感哈希,并带有 Python 绑定。
  • kampersanda/mih-rs - Rust 实现的多索引哈希算法,用于在汉明空间中对 64 位编码进行近邻搜索。

强化学习

监督学习模型

非监督学习与聚类模型

统计模型

进化算法

参考

周边项目

博客

引言

教程

应用

案例研究

讨论

图书

影片

播客

论文

  • Rust 中的端到端 NLP 流程,第二届 NLP 开源软件研讨会(NLP-OSS)论文集,第20–25页,虚拟会议,2020年11月19日,Guillaume Becquin

如何贡献

请直接更新 README.md 文件。

一旦您更新了此 README.md 文件,CI 将会自动执行, 同时网站内容也会相应更新。

感谢

感谢所有相关项目。

https://github.com/vaaaaanquish/Awesome-Rust-MachineLearning

相似工具推荐

openclaw

OpenClaw 是一款专为个人打造的本地化 AI 助手,旨在让你在自己的设备上拥有完全可控的智能伙伴。它打破了传统 AI 助手局限于特定网页或应用的束缚,能够直接接入你日常使用的各类通讯渠道,包括微信、WhatsApp、Telegram、Discord、iMessage 等数十种平台。无论你在哪个聊天软件中发送消息,OpenClaw 都能即时响应,甚至支持在 macOS、iOS 和 Android 设备上进行语音交互,并提供实时的画布渲染功能供你操控。 这款工具主要解决了用户对数据隐私、响应速度以及“始终在线”体验的需求。通过将 AI 部署在本地,用户无需依赖云端服务即可享受快速、私密的智能辅助,真正实现了“你的数据,你做主”。其独特的技术亮点在于强大的网关架构,将控制平面与核心助手分离,确保跨平台通信的流畅性与扩展性。 OpenClaw 非常适合希望构建个性化工作流的技术爱好者、开发者,以及注重隐私保护且不愿被单一生态绑定的普通用户。只要具备基础的终端操作能力(支持 macOS、Linux 及 Windows WSL2),即可通过简单的命令行引导完成部署。如果你渴望拥有一个懂你

349.3k|★★★☆☆|昨天
Agent开发框架图像

stable-diffusion-webui

stable-diffusion-webui 是一个基于 Gradio 构建的网页版操作界面,旨在让用户能够轻松地在本地运行和使用强大的 Stable Diffusion 图像生成模型。它解决了原始模型依赖命令行、操作门槛高且功能分散的痛点,将复杂的 AI 绘图流程整合进一个直观易用的图形化平台。 无论是希望快速上手的普通创作者、需要精细控制画面细节的设计师,还是想要深入探索模型潜力的开发者与研究人员,都能从中获益。其核心亮点在于极高的功能丰富度:不仅支持文生图、图生图、局部重绘(Inpainting)和外绘(Outpainting)等基础模式,还独创了注意力机制调整、提示词矩阵、负向提示词以及“高清修复”等高级功能。此外,它内置了 GFPGAN 和 CodeFormer 等人脸修复工具,支持多种神经网络放大算法,并允许用户通过插件系统无限扩展能力。即使是显存有限的设备,stable-diffusion-webui 也提供了相应的优化选项,让高质量的 AI 艺术创作变得触手可及。

162.1k|★★★☆☆|2天前
开发框架图像Agent

everything-claude-code

everything-claude-code 是一套专为 AI 编程助手(如 Claude Code、Codex、Cursor 等)打造的高性能优化系统。它不仅仅是一组配置文件,而是一个经过长期实战打磨的完整框架,旨在解决 AI 代理在实际开发中面临的效率低下、记忆丢失、安全隐患及缺乏持续学习能力等核心痛点。 通过引入技能模块化、直觉增强、记忆持久化机制以及内置的安全扫描功能,everything-claude-code 能显著提升 AI 在复杂任务中的表现,帮助开发者构建更稳定、更智能的生产级 AI 代理。其独特的“研究优先”开发理念和针对 Token 消耗的优化策略,使得模型响应更快、成本更低,同时有效防御潜在的攻击向量。 这套工具特别适合软件开发者、AI 研究人员以及希望深度定制 AI 工作流的技术团队使用。无论您是在构建大型代码库,还是需要 AI 协助进行安全审计与自动化测试,everything-claude-code 都能提供强大的底层支持。作为一个曾荣获 Anthropic 黑客大奖的开源项目,它融合了多语言支持与丰富的实战钩子(hooks),让 AI 真正成长为懂上

143.9k|★★☆☆☆|今天
开发框架Agent语言模型

ComfyUI

ComfyUI 是一款功能强大且高度模块化的视觉 AI 引擎,专为设计和执行复杂的 Stable Diffusion 图像生成流程而打造。它摒弃了传统的代码编写模式,采用直观的节点式流程图界面,让用户通过连接不同的功能模块即可构建个性化的生成管线。 这一设计巧妙解决了高级 AI 绘图工作流配置复杂、灵活性不足的痛点。用户无需具备编程背景,也能自由组合模型、调整参数并实时预览效果,轻松实现从基础文生图到多步骤高清修复等各类复杂任务。ComfyUI 拥有极佳的兼容性,不仅支持 Windows、macOS 和 Linux 全平台,还广泛适配 NVIDIA、AMD、Intel 及苹果 Silicon 等多种硬件架构,并率先支持 SDXL、Flux、SD3 等前沿模型。 无论是希望深入探索算法潜力的研究人员和开发者,还是追求极致创作自由度的设计师与资深 AI 绘画爱好者,ComfyUI 都能提供强大的支持。其独特的模块化架构允许社区不断扩展新功能,使其成为当前最灵活、生态最丰富的开源扩散模型工具之一,帮助用户将创意高效转化为现实。

107.9k|★★☆☆☆|昨天
开发框架图像Agent

markitdown

MarkItDown 是一款由微软 AutoGen 团队打造的轻量级 Python 工具,专为将各类文件高效转换为 Markdown 格式而设计。它支持 PDF、Word、Excel、PPT、图片(含 OCR)、音频(含语音转录)、HTML 乃至 YouTube 链接等多种格式的解析,能够精准提取文档中的标题、列表、表格和链接等关键结构信息。 在人工智能应用日益普及的今天,大语言模型(LLM)虽擅长处理文本,却难以直接读取复杂的二进制办公文档。MarkItDown 恰好解决了这一痛点,它将非结构化或半结构化的文件转化为模型“原生理解”且 Token 效率极高的 Markdown 格式,成为连接本地文件与 AI 分析 pipeline 的理想桥梁。此外,它还提供了 MCP(模型上下文协议)服务器,可无缝集成到 Claude Desktop 等 LLM 应用中。 这款工具特别适合开发者、数据科学家及 AI 研究人员使用,尤其是那些需要构建文档检索增强生成(RAG)系统、进行批量文本分析或希望让 AI 助手直接“阅读”本地文件的用户。虽然生成的内容也具备一定可读性,但其核心优势在于为机器

93.4k|★★☆☆☆|昨天
插件开发框架

LLMs-from-scratch

LLMs-from-scratch 是一个基于 PyTorch 的开源教育项目,旨在引导用户从零开始一步步构建一个类似 ChatGPT 的大型语言模型(LLM)。它不仅是同名技术著作的官方代码库,更提供了一套完整的实践方案,涵盖模型开发、预训练及微调的全过程。 该项目主要解决了大模型领域“黑盒化”的学习痛点。许多开发者虽能调用现成模型,却难以深入理解其内部架构与训练机制。通过亲手编写每一行核心代码,用户能够透彻掌握 Transformer 架构、注意力机制等关键原理,从而真正理解大模型是如何“思考”的。此外,项目还包含了加载大型预训练权重进行微调的代码,帮助用户将理论知识延伸至实际应用。 LLMs-from-scratch 特别适合希望深入底层原理的 AI 开发者、研究人员以及计算机专业的学生。对于不满足于仅使用 API,而是渴望探究模型构建细节的技术人员而言,这是极佳的学习资源。其独特的技术亮点在于“循序渐进”的教学设计:将复杂的系统工程拆解为清晰的步骤,配合详细的图表与示例,让构建一个虽小但功能完备的大模型变得触手可及。无论你是想夯实理论基础,还是为未来研发更大规模的模型做准备

90.1k|★★★☆☆|昨天
语言模型图像Agent