高级编译# 注意 这是新版 JAX 教程草案 中一个章节的占位符。 目前,您可以在旧版文档中找到一些相关内容。 提前降低和编译 使用 C++ 和 CUDA 为 GPU 创建自定义操作 Pallas: JAX 内核语言