TensorPlay AI

生态系统

从理论理解到实际实现,所有层都可以协同工作,也可以独立拆开。

P10

基础张量计算引擎,覆盖内存、算子和硬件后端。

TPX

面向动态图的显式自动微分层。

Stax

静态图优化、JIT、算子融合与图捕获实验场。

NN

PyTorch 兼容的高层神经网络 API。

从学习到硬件

Pythonic 接口、可读 CPU 与 CUDA 后端,以及 nanobind 集成,让 TensorPlay 同时适合深度学习基础学习、自定义网络层和硬件内核实验。

Ask DeepWiki