nevermind-neu

用 Rust 编写的带终端用户界面的机器学习库和工具。它支持 OpenCL 层和 CPU 层。核心数学矩阵库是 ndarray，它使用 matrixmultiply 包进行 CPU 矩阵乘法。

设计目标

• 快速优化的计算
• 用户友好的 API
• 提供用户友好的终端应用程序以创建、训练和管理模型

OpenCL

OpenCL 支持基于 ocl 包。这是一个可选功能，默认启用。

功能

• 全连接层
• 欧几里得损失，softmax 损失
• 优化器：Adam、RMSProp、AdaGrad、AdaDelta
• 异步并行数据加载
• Protobuf、CSV 数据加载器
• 将神经网络状态序列化为 protobuf
• 将神经网络配置序列化为 net yaml 文件
• 激活函数：sigmoid、tanh、relu、leaky_relu

终端用户界面工具

示例

XOR 函数逼近

cargorun --release --examplexor

XOR 函数逼近（带 OpenCL 支持）

cargorun --release --examplexor_ocl

有关 MNIST 数字分类示例，请参阅 MNIST 示例说明

路线图

• Conv2D 层
• RNN + LSTM
• 残差块
• OpenCL 优化

许可

Apache 许可协议版本 2.0