CustomCall 失败: jaxlib/gpu/prng_kernels

在 Google Colab 中运行时，我收到以下错误：我使用的是 Google Colab 的 Pro 版本。