tch-rs

tch-rs 是一个为 Rust 语言提供 PyTorch C++ API 绑定的库。该项目通过薄层封装使得 Rust 开发者能够直接调用 PyTorch 的功能，同时尽量保持与原生 C++ API 一致，便于熟悉 PyTorch 的用户快速上手。它支持深度学习模型的训练与推理，并且可以利用 GPU 加速计算。适用于需要结合 Rust 的高效性能和 PyTorch 强大机器学习能力的应用场景，如高性能计算、嵌入式系统中的AI应用等。此外，项目提供了详尽的文档和支持多种安装方式，方便开发者根据自身环境灵活选择。