开源项目
mattn/tensai:纯Go小型机器学习框架
mattn/tensai
概述
tensai是mattn发布的Go小型机器学习框架,用于学习与实验:纯Go实现前向传播、反向传播与优化,默认构建无外部依赖(可选wgpu构建标签仅引入一个无cgo依赖)。功能包括矩阵与N维张量运算(NumPy式广播、批量MatMul及其转置形式、Reshape与零拷贝视图),以及基于Go实验性simd/archsimd包的SIMD加速:amd64上的AVX2与arm64上的NEON内核覆盖Matmul、ReLU、Softmax、GELU、LayerNorm与Adam更新等,仍保持纯Go、无cgo、无汇编文件。SIMD需GOEXPERIMENT=simd并配合Go 1.26/1.27。仓库于2026-08-19以仓库创建形式发布,提供英文与日文文档。