tch-rs

收录理由

对于希望在 Rust 中直接使用 PyTorch 能力的团队，这个库提供了围绕 libtorch C++ API 的轻量封装，忠实于上游接口。你可以像写普通 Rust 代码一样调用张量操作、自动微分、SGD 和 Adam 等优化器，以及熟悉的 nn 模块来组合网络层。由于绑定与 C++ 接口高度一致，PyTorch 的官方文档和示例基本可以照搬，行为也与上游保持同步。安装方式灵活多样：可以链接系统自带的 libtorch，也可以通过环境变量指向 Python 的 torch 安装，或者让构建脚本自动下载预编译的 CPU 或 CUDA 版本。这种弹性既能满足需要快速矩阵运算的生产服务，也适合完全在 Rust 中尝试自定义架构的小团队。