Rust 机器学习,一定要知道的这些库,可以替代 Python 库了

2024年 4月 26日 101.7k 0

大家好,我是渔夫。

今天聊聊,Rust 机器学习必须了解的库。

最近,很多伙伴在机器学习领域,将自己主力语言 Python 开始瞄准到 Rust。头疼的是,Rust 还是太年轻了,很多库要么还没成熟,要么还在开发中,没有 Python 那么丰富。

不过,Rust 社区非常活跃并且增长迅猛,很多大佬也在努力尝试 Rust 来开发很多机器学习库,但目前为止,很多大厂积极拥抱 Rust 来重构底层基础设施,这是值得关注。下面是,Rust 常见机器学习库。

ndarray

ndarray 是 NumPy 库中的一个核心数据结构,它是一个用于表示多维数组的对象。ndarray 在存储和操作大型多维数组时非常高效,广泛应用于科学计算、数据分析、机器学习等领域。

在 Rust 开发中,ndarray 库可以作为替代  NumPy 库。

Github 地址:https://github.com/rust-ndarray/ndarray

Polars

可以作为替代 Python 中 pandas 库,Polars 是个开源的数据处理库,提供快速和灵活的数据处理能力,尤其处理大型数据集时。Polars 的设计目标是提供类似于 Pandas 的 API,但利用 Rust 的性能优势来提高数据处理的速度。

Github 地址:https://github.com/pola-rs/polars

tch-rs

tch-rs 是一个使得 Rust 能够与 PyTorch 框架结合的工具,扩展了 Rust 在机器学习和深度学习领域的应用能力。广泛用于计算机视觉和自然语言处理等研究和应用领域,tch-rs 允许 Rust 开发者利用 PyTorch 的功能,进行深度学习模型的开发和部署。

Github 地址:https://github.com/LaurentMazare/tch-rs

Burn

Burn 是一个新的综合动态深度学习框架,使用 Rust 构建,以极高的灵活性、计算效率和可移植性作为其主要目标。这个库之前也推荐过,非常值得去关注。

Github 地址:https://github.com/tracel-ai/burn

Candle

Candle 是由 Hugging Face 开发的 Rust 语言机器学习框架。主要是为 Rust 开发者在机器学习领域提供了新的工具,使得构建和部署机器学习应用变得更加容易。

  • 极简设计:提供简洁的 API,易于使用。
  • 高性能:利用 Rust 语言的性能优势。
  • 易用性:API 设计借鉴了 PyTorch,降低学习成本。
  • 灵活性:支持多种后端,方便在不同环境中部署。
  • 安全性:Rust 语言保证了内存安全和线程安全。

GIthub 地址:https://github.com/huggingface/candle

Linfa

linfa 提供一个全面的工具包来使用 Rust 构建机器学习应用程序。类似 Python 的scikit-learn 库,专注于日常 ML 任务的常见预处理任务和经典 ML 算法。

GIthub 地址:https://github.com/rust-ml/linfa

plotters

Plotters 是一个绘图库,设计用于用纯 Rust 渲染图形、绘图和图表,支持各种类型的后端,包括位图、矢量图、活塞窗口、GTK/Cairo 和 WebAssembly。类似 Python 的 matplotlib 库。

Github 地址:https://github.com/plotters-rs/plotters

petgraph

petgraph 是一个图数据结构库,图是节点以及节点之间的边的集合。petgraph 提供了多种图形类型、这些图形的算法以及以格式输出图形 graphviz 的功能。

Github 地址:https://github.com/petgraph/petgraph

tensorflow-rust

TensorFlow 的 Rust 语言绑定,允许 Rust 开发者使用 TensorFlow 的机器学习框架。

Github 地址:https://github.com/tensorflow/rust

相关文章

JavaScript2024新功能:Object.groupBy、正则表达式v标志
PHP trim 函数对多字节字符的使用和限制
新函数 json_validate() 、randomizer 类扩展…20 个PHP 8.3 新特性全面解析
使用HTMX为WordPress增效:如何在不使用复杂框架的情况下增强平台功能
为React 19做准备:WordPress 6.6用户指南
如何删除WordPress中的所有评论

发布评论