Hugging Face··12 天前
Hugging Face ❤️ TorchTPU:原生支持 TPU
Hugging Face ❤️ TorchTPU: TPUs, Natively
博客文章,介绍 TorchTPU 这一原生 PyTorch 后端,使 TPU 成为 PyTorch 中的普通设备;文中称 Transformers、Diffusers 和 Kernels 可在 TPU 上原生运行,从 Hub 选择模型后发送到 "tpu" 即可。文章还简介了 TPU 作为 Google 自研 AI 加速器的情况与可用规模,并对比了此前需经 torch_xla 的旧方式及其限制。面向希望在 TPU 上运行 PyTorch 模型的用户。
- 所属主题
- 模型服务与算力
- 收录时间
- 2026-10-26
推荐阅读
PyTorch 2.15 中的 MPS
MPS in PyTorch 2.15
Hugging Face PyTorch 团队撰写的博客文章,介绍 PyTorch 2.15 中 Apple MPS 后端的更新,包括 MPS 支持 FP8、语言模型解码在线性层与 softmax 上获得更快路径、以及针对音频、视频和训练工作负载中新 Metal 内核。文中说明该团队以 Hugging Face Hub 上的真实模型排查 MPS 不足并向上游修复,并提到 MoE 模型解码速度较 2.14 提升 5–6.5 倍。适合在 Mac 上运行或训练 PyTorch 模型的读者。
TorchNeuron:在 AWS Trainium 上运行 HF 模型前所未有地简单
TorchNeuron: running HF models on AWS Trainium is easier than ever
这是 Hugging Face 博客的一篇文章,介绍为 AWS Trainium 提供 PyTorch 原生后端的 TorchNeuron 插件,支持 eager 模式、torch.compile 以及 FSDP、TP 等原生分布式 API。文章说明 Transformers、Diffusers 和 Kernels 已原生支持该后端,此前 HF 在 Trainium 上依赖基于 PyTorch XLA 的 optimum-neuron;TorchNeuron 自 2026 年 10 月 20 日起进入公开测试,可通过 pip 安装。
StartLuxLabs/StartLux-Decision
这是一个开源决策模型系列的项目仓库,模型规模从 0.8B 到 27B,一次推理即为每个选项给出校准概率,仅在不确定时输出简短思考。项目提供面向 vLLM、llama.cpp 和 MLX 的推理引擎,并提供 Hugging Face 模型页、arXiv 技术报告、API 文档与博客链接。
你的模型能在 AWS Trainium 上运行吗?查一查就知道
Does Your Model Run on AWS Trainium? Look It Up
文章介绍一个 torch-neuronx 模型支持情况 Space,用于查询 Hub 上热门模型能否在 AWS Trainium 上运行。该 Space 的 Selected 视图展示按下载量和趋势分挑选的模型,每行给出结果、模型分片占用的 Trainium 核心数、运行实际耗时(含编译)和峰值设备内存;Trending 和 Downloads 视图按 Hub 自身排名逐任务列出结果,并对故意未测试的模型说明原因,还可选择 neuronx-cc 和 Transformers 的版本。这是继此前介绍 TorchNeuron 的文章之后的更新。
cobanov/PS5LM
该项目把 llama.cpp 移植到已越狱的 PlayStation 5 上,允许主机原生运行 GGUF 格式模型,从 Qwen 3.8 开始。README 说明它借助主机 GPU 推理,提供 OpenAI 与 Anthropic 兼容 API 和适配电视的界面,可在无 PC、无云端、无账号的情况下为局域网内设备提供模型服务,另新增了 Ollaya 的决策模型支持。适合拥有越狱 PS5、想在家搭私有模型服务的读者。
为一万到一万二千美元预算的本地 agentic coding 装机:1× 5090 对比 4× 3090、混合方案、PRO 6000、Mac Studio?
$10–12K build for local agentic coding: 1× 5090 vs 4× 3090 vs hybrid vs PRO 6000 vs Mac Studio?
r/LocalLLaMA 社区帖子。发帖者准备花约 1 万美元(必要时到 1.2 万)买硬件做本地 agentic coding,用途包括 agentic coding、偶尔 2–4 个并行 subagent 和逆向工程,可选需求是用 Hunyuan3D-2.1 做贴图生成(需 CUDA 和约 21GB 显存)。其列出正在考虑的方案:单张 RTX 5090、四张二手 3090、5090 + 2×3090 混合、Mac Studio M5 Ultra 256GB、超预算的 PRO 6000,并就长上下文中单 5090 与四 3090 的日常体验、3090 长上下文预填充表现、混合方案在 AM5 上以 x8/x8/x4 跑两个 vLLM 实例是否值得、以及是否该等 50 Super 或改装 4090 48GB 等问题征询意见。适合正在规划本地多卡推理/编码 agent 装机的人阅读。