研究者业余将主流视频生成模型从 PyTorch 移植到 Jax 适配 TPU
ceciletamura · x · 2026-09-24
研究者 CongyueD 发布了一个业余开发的 TPU 库，将大多数视频生成模型从 PyTorch 翻译为 Jax，以便在 Google TPU 上运行。项目附有源代码、技术报告，并已上架 PyPI 可快速试用，作者感谢 Google TPU Research Cloud（TRC）提供的算力支持。对想在 TPU 生态跑视频模型的研究者是个现成工具。
「Infra」频道最新
- ComfyUI 无损压缩节点：28GB 模型压到 19GB，位级完全一致 — New-Shift6661 · 2026-09-24
- 跑 LLM 别只看核心数：AMD CCD 数量决定带宽上限 — HankYeomans · 2026-09-24
- 十亿人用个人 AI Agent,算力与内存需求将何等惊人 — firstadopter · 2026-09-24
- Cloudflare 人机验证被吐槽浪费爬虫与 Agent 的时间和 token — msg · 2026-09-24
- DuckDB 首次内置进 dbt v2,基于 Rust Fusion 引擎发布 — josh_wills · 2026-09-24
- 马斯克晒首批自产镍正极 Cybercab,美洲首座正极厂落地得州 — elonmusk · 2026-09-24