dfdx
dfdx是一个注重人体工学和安全性的Rust深度学习库,支持GPU加速和最多6维的张量形状。它在编译时进行形状和类型检查,提供多种张量操作,例如矩阵乘法和卷积。该库还包含神经网络构建模块和标准的深度学习优化器,如Sgd和Adam。设计目标是性能最大化和最小化不安全代码。用户可以启用CUDA特性进行GPU加速,非常适合在Rust中进行深度学习开发的用户。