grenade
Grenade 是一个高效实用的递归神经网络库,专为 Haskell 语言设计,支持复杂网络的简洁精确定义。几行代码就能指定并初始化一个在 MNIST 数据集上达到约1.5%测试误差的神经网络。Grenade 支持卷积、池化、全连接、LSTM 等多种层类型,内置反向传播和梯度更新功能。基于纯函数设计,允许灵活组合训练函数,甚至实现生成对抗网络。性能依托 hmatrix、BLAS 和 LAPACK,支持并行处理。