tch-rs

tch-rs 是一个为 Rust 提供 PyTorch C++ API（libtorch）绑定的开源库。它通过轻量级封装直接对接 libtorch，保持与原始 C++ 接口高度一致，支持 CPU 和 CUDA 后端，并提供自动下载预编译二进制、环境变量配置及 Python PyTorch 复用等多种部署方式。项目不抽象高层模型接口，而是作为底层系统级桥梁，便于在 Rust 生态中安全、高效地执行张量计算、自动微分和神经网络推理/训练。适用于需将 PyTorch 模型集成至 Rust 服务（如低延迟推理服务、嵌入式 AI 应用或高性能系统组件）的场景。