PyTorch

22 篇报道

PyTorch 是 Meta 维护的开源深度学习框架,以 Python 张量和动态神经网络为核心,提供 GPU 加速。近期报道集中在后端与性能上:PyTorch 2.15 为 Mac MPS 带来 FP8,MoE 解码快 5–6.5 倍,Qwen3.6-35B-A3B 在 M3 Ultra 上从 9.4 提升到 61.3 tokens/s;Hugging Face 则推出 TorchTPU 与 TorchNeuron,让 Transformers 和 Diffusers 直接跑在 TPU 与 AWS Trainium 上。此外还有从零实现 transformer 手动反向传播、GPT-2 KV cache 等教程类内容。

相关话题

GPU内核开发环境搭建实用要点

一篇r/MachineLearning帖子汇总了GPU内核开发环境搭建中常被忽视的问题:CUDA 13已放弃Maxwell、Pascal和Volta架构,需按计算能力选装工具链;PyTorch自带CUDA运行时,不能说明nvcc可用;GPU启动为异步,需预热、同步并用CUDA事件计时,否则测的是入队时间。

r/MachineLearning ·

从零实现GPT-2 KV缓存并测量性能

一位开发者用PyTorch从零实现了GPT-2 small的KV缓存,并测量了延迟和内存。在上下文长度1000时,缓存将每token时间从29.69毫秒降至4.95毫秒,生成1016个token的总时间从14.66秒降至5.03秒。缓存与未缓存的贪心生成结果在200个token内一致。

r/MachineLearning ·

pytorch

支持 GPU 加速的 Python 张量与动态神经网络库。

▲ 315 · GitHub Trending ·

pytorch

Python 张量和动态神经网络,具有强大的 GPU 加速。

▲ 84 · GitHub Trending ·

PyTorch

支持强GPU加速的Python张量与动态神经网络库。

▲ 81 · GitHub Trending ·
已经到底了