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