跳到正文
Google Developers Blog· noreply@aihot.news (Google Developers Blog)·· 1 天前精选

Google 发布 TorchTitan TPU 分叉并给出模型优化 Playbook

AI 导读

Google 推出 TorchTitan TPU 分叉(github.com/google-pytorch/torchtitan),配合 TorchTPU 原生后端在 Cloud TPU 上训练 PyTorch 模型,提供 SimpleFSDP、Graph Trainer、Pallas 内核和优化 Playbook。

推荐理由

来源于 AI Hot 精选订阅

来源:Google Developers Blog · developers.googleblog.com