gotch

> 编程语言

开源

用于 Pytorch C++ API (libtorch) 的 Go 绑定

652 stars0 点赞1 次浏览

访问官网 GitHub

工具介绍

用于 Pytorch C++ API (libtorch) 的 Go 绑定

Gotch() (https://opensource.org/licenses/Apache-2.0)[](https://pkg.go.dev/github.com/sugarme/gotch?tab=doc)

概述

gotch 创建了一个对 Pytorch C++ API (Libtorch) 的薄层包装,以利用其已经优化的 C++ 向量 API (3039) 和支持 CUDA 的动态图形计算,并提供了 Go 语言的标准 API,用于在 Go 中开发和实现深度学习。

某些功能尚未实现 - [x] 全面的 Pytorch 张量 API (2525) - [x] 完整的 Pytorch 动态图形计算 - [x] JIT 接口,可运行使用 PyTorch Python API 训练/保存的模型 - [x] 加载预训练的 Pytorch 模型并进行推理 - [x] 纯 Go API,用于构建和训练神经网络模型,支持 CPU 和 GPU - [x] 最新的图像模型 - [ ] NLP 语言模型 - Transformer,使用 gotch 和纯 Go 标记器构建的单独包 - gotch 处于活动开发模式,可能会发生 API 更改。欢迎 Pull Request、报告问题或讨论任何疑问。所有贡献都受欢迎。

gotch 当前版本为 v0.9.1