MiniTensor：轻量级高性能张量运算库
机器学习
2026-02-03 v1 人工智能
数学软件
摘要
我们提出了 MiniTensor，一个专注于极简主义、正确性和性能的开源张量运算库。MiniTensor 提供类似 PyTorch 的熟悉 Python API，同时在性能关键代码中采用 Rust 引擎执行。核心支持稠密 n 维张量、广播、约化、矩阵乘法、逆向模式自动微分、一套紧凑的神经网络层以及标准优化器。本文描述了 MiniTensor 架构的设计，包括其高效内存管理、用于梯度的动态计算图以及通过 PyO3 与 Python 的集成。我们还将 MiniTensor 的安装占用空间与 PyTorch 和 TensorFlow 进行比较，表明 MiniTensor 仅占用几兆字节的包体积，比主流框架小几个数量级，同时保留了在 CPU 上进行科研和开发所必需的基本功能。该仓库可在 https://github.com/neuralsorcerer/minitensor 找到。
关键词
引用
@article{arxiv.2602.00125,
title = {MiniTensor: A Lightweight, High-Performance Tensor Operations Library},
author = {Soumyadip Sarkar},
journal= {arXiv preprint arXiv:2602.00125},
year = {2026}
}