piston

概念验证性质的库和网页试验场,利用WebGPU在浏览器中训练Transformer、LSTM等小型序列模型,用于学习和实验。

分享到 X许可证: MIT

概述

Piston 是运行在 WebGPU 上的概念验证型自动微分库,Sequence Toy 是基于它构建的网页试验场。两者结合,可以直接在浏览器中训练 Transformer、LSTM、GRU 和普通 RNN 等小型序列模型。Piston 是在 Ratchet 基础上加入自动微分的分支,更适合作为学习和实验项目。

主要特点

  • 用 WebGPU 在浏览器中训练序列模型
  • 支持 Transformer、LSTM、GRU 和普通 RNN
  • 用 Rust 编写的自动微分库
  • 附带 Sequence Toy 网页试验场

适用场景

适合想在浏览器里试验训练小型语言模型的人。作者称 Piston 只是概念验证,不应期待它是生产级框架。

上游仓库
vinhowe/piston
GitHub 上的 fork
Guo-astro/piston
上游 Star 数
36
分类
AI 智能体与大模型工具
编程语言
Rust
许可证
MIT
Fork 时间
2025-11-21
同步状态
已同步上次同步 2026-09-29