提到机器学习领域,大部分人第一时间想到了Python,相对于它成熟的生态和丰富的库,当然不可否认,它确实还是领域主导语言。
但是Rust 因其安全性、性能和并发特性而在机器学习 (ML) 和人工智能 (AI) 领域受到关注。马斯克也在推特公开承认,Rust才是AGI语言。虽然Rust还在积极的发展中,5月的编程排行榜也不过还是18名,但他的生态正在逐步发展,作为一个Rust爱好者,我始终相信他就是未来。
本文的重点在于讨论Rust中两个比较流行的机器学习框架,而不是歌颂Rust,也没有任何语言歧视,自然也不会试图无脑的劝各位入坑Rust。
🎂Candle
一个简单且高性能的深度学习框架。
这是一个以简单和高性能为主打而设计的速度学习框架。它提供了一个用于定义个训练神经网络的简约API,利用基于强大的内核的并行计算功能。
实现上,Candle底层基于cuTENSOR和cuDNNv8库,这些库可以在NVIDIA的CPU上高效执行。下面是基于Candle定义和训练MLP的简单代码示例:
use candle::prelude::*;
fn main() {
let mut model = MlpBuilder::new(vec![784, 128, 64, 10])
.with_activation(Activation

最低0.47元/天 解锁文章
583

被折叠的 条评论
为什么被折叠?



