提到机器学习领域,大部分人第一时间想到了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