JAX CPU 回退导致 Erdős 问题出现超时问题

你好，对于 Erdős 问题，默认情况下会运行 JAX。如果没有安装 CUDA 版本的 jaxlib，则会回退到 CPU 版本的 JAX。在这种情况下，特别是对于较小的模型，执行通常会触及超时限制，我认为这可能会影响性能。即使我们提高超时限制，似乎仍然存在实际限制。考虑到这一点，我正在考虑安装支持 CUDA 的 JAX 版本，并进行最多 200 次迭代。我很希望您对这种方法有所建议。