演讲中更有趣的部分之一是特斯拉的 Dojo 团队在其超级计算机上取得的进展。虽然头条新闻主要集中在特斯拉为其 Dojo 超级计算机开发的硬件上,但该公司在软件方面也取得了一些重大进展。
特斯拉首席工程师 Rajiv Kurian 目前专注于微架构的硬件-软件协同设计,他在演讲中解释了特斯拉如何通过软件充分利用其 Dojo 超级计算机 AI 加速器。该演示文稿详细介绍了特斯拉如何开发其堆栈以使 Dojo 编译器能够快速运行开箱即用的神经网络。
Kurian 提供了一个案例:“我们希望大多数模型都能开箱即用。例如,我们采用了最近发布的 Stable Diffusion 模型,并在几分钟内让它在 Dojo 上运行。开箱即用,编译器能够以模型并行方式在 25 个 Dojo 模具中进行映射”。
Stable Diffusion 模型是一种文本到图像的深度学习模型,能够生成以文本描述为条件的详细图像。他发布了一些由 Stable Diffusion 生成的“Cybertruck on Mars”图片:
扫一扫
在手机上阅读