研究者寻求 TPU 访问权限，用于 Transformer 工作负载的 ML 映射工具验证

Need access to TPUs for ML mapping project [R]

AI 导读

一位研究者正在开发面向 Transformer 工作负载的分析映射/编译工具，可将负载映射到采用软件管理暂存器的硬件平台，并已在多家初创公司和大型公司的芯片上验证其映射器在多种工作负载下优于这些厂商的软件栈。

正文

I am working on a project where I am building a analytical mapping/compilation tool for transformer workloads into hardware platforms that have software managed scratchpads.

I got hold of a few chips from some startups and bigger companies and have shown that my mapper beat their software stack for a variety of workloads.

But I would like to show that it also beats a more mature software stack like Google's. And for that I would need access to TPUs. I already applied for their TPU research cloud program a month ago but still did not hear from them.

Anybody know of a somewhat affordable way to get access to their TPU? I will most likely need to use one for about 10 hours total or so.

submitted by /u/Green_Rays
[link] [留言]