介绍Pallas在Keras中编写自定义TPU/GPU内核

你不再需要离开Python来编写高性能硬件内核。

了解如何在Keras中使用Pallas来编写自定义操作，这些操作可以降级为用于TPU的Mosaic或用于GPU的Triton：