Skip to content

TensorPlay透明的 AI 架构

一个适合学习者、兼容 PyTorch 的深度学习框架,旨在揭示神经网络内部原理并促进自定义硬件实验。

TensorPlay LogoTensorPlay Logo

本地开始

选择您的偏好并运行安装命令

TensorPlay 版本
包管理器
计算平台
运行此命令:已复制!
pip install tensorplay --index-url https://download.tensorplay.cn/whl/cu130

TensorPlay 生态系统

一系列设计用于协同工作或独立使用的库。

P10

核心计算引擎。提供内存管理和基础张量内核的干净、可读实现。

设计理念 →

TPX

显式自动微分层。专为想要理解或修改计算图构建方式的人设计。

设计理念 →

Stax

优化游乐场。在简化环境中试验算子融合和静态图捕获。

设计理念 →

NN

模块化业务层。高级组件 (Linear, Conv2d),作为你自己网络层的蓝图。

设计理念 →

最新动态

关注我们的最新博文和研究亮点。

TensorPlay 架构设计:为什么选择四个核心库?

深入探讨 TensorPlay “解耦优先”的设计哲学。通过将框架拆分为 P10、TPX、Stax 和 NN 四个独立且职责单一的库,我们实现了灵活性与高度可定制性的完美平衡。

深入 P10 核心:Dispatcher 与硬件抽象层

揭秘 TensorPlay 如何在底层实现跨硬件的算子调度。通过 Dispatcher 机制与 TensorImpl 模式,P10 能够以极低的延迟将计算任务分发至最合适的硬件后端。

TPX 自动微分:如何实现计算与梯度的完美解耦?

传统的 autograd 往往深度嵌入计算逻辑。TPX 采用了一种全新的“透明扩展”模式,在完全不侵入 P10 计算逻辑的前提下,实现了灵活的动态图追踪与梯度回传。

基于 Apache 2.0 许可发布。

📚DeepWiki