Pallas已经取得了长足的进步。

François Chollet

您无需离开Python即可编写高性能硬件内核。

了解如何在Keras中使用Pallas编写自定义操作，这些操作可编译为TPU的Mosaic或GPU的Triton：
https://t.co/oeV4cmV4M0

打开互动版