Pallas

Pallas DSL 助力 Google TPU 自定义内核优化

本指南介绍了 Pallas，一个基于 Python 的领域特定语言 (DSL)，用于编写可为 Google 的张量处理单元 (TPU) 编译和优化的自定义内核。Pallas 旨在为开发人员提供对硬件操作的细粒度控制，使他们能够获得超越标准库的性能提升。本文档是了解 Pallas 功能及其在 TPU 上优化机器学习工作负载方面作用的入门资源。