# tinytorch

**Repository Path**: xntk2381/tinytorch

## Basic Information

- **Project Name**: tinytorch
- **Description**: No description available
- **Primary Language**: Go
- **License**: MIT
- **Default Branch**: master
- **Homepage**: None
- **GVP Project**: No

## Statistics

- **Stars**: 0
- **Forks**: 0
- **Created**: 2026-01-07
- **Last Updated**: 2026-01-27

## Categories & Tags

**Categories**: Uncategorized

**Tags**: None

## README



# 🧠 TinyTorch-Go  
*一个极简、教学导向的深度学习库（Go 语言实现）—— 灵感源自 [TinyTorch](https://mlsysbook.ai/tinytorch)*  

[![Go Version](https://img.shields.io/badge/Go-%3E%3D1.22-blue)](https://golang.org)  
[![License](https://img.shields.io/badge/License-MIT-green)](LICENSE)

> 🔍 **目标**：深入理解 PyTorch 等现代张量库的底层原理——**亲手用 Go 从零构建**。  
> ✅ 覆盖：张量（Tensor）、自动微分（Autograd）、计算图、基础神经网络层，以及简易训练循环。

---

## 📦 核心功能

| 模块 | 状态 | 说明 |
|------|------|------|
| `Tensor` | ✅ 已完成 | 支持 N 维数组，含 shape、data；基础运算（`Add`, `Mul`, `Matmul`） |


> 🚧 *规划中*：`Conv2D`、`CrossEntropyLoss`、数据加载器（DataLoader）、GPU 支持（CGO/cuBLAS）、ONNX 导出。

---

## 🚀 快速上手

### 1. 克隆 & 初始化
```bash
git clone https://gitee.com/xntk2381/tinytorch.git
cd tinytorch
go mod tidy
```


---

## 📖 学习资源

本项目实现思路参考：
- 📜 [Module 01: Tensor](https://mlsysbook.ai/tinytorch/modules/01_tensor_ABOUT.html)

---

## 🤝 贡献指南

欢迎提交 PR！尤其欢迎以下方向：
- ✅ 新增运算（`Tanh`, `Softmax`, `Conv1D`）
- 📈 性能优化（strided 视图、内存池）
- 🧪 更多测试与示例（CNN、RNN）
- 📚 教程文档（`docs/` 目录）

**贡献流程**：  
1. Fork 仓库 → 2. 创建特性分支（`feat/xxx`）→ 3. 提交代码 → 4. 提交 PR（附说明）

---

## 📜 开源协议

MIT 许可证 — 详见 [LICENSE](LICENSE)

---

## 🙏 致谢

- 感谢《[机器学习系统](https://mlsysbook.ai/book/)》作者的精妙教学设计

---

> 🚀 **不仅要会用框架，更要理解系统**。  

祝您开发顺利，学有所得！ 💙