我正在寻找具有 GPU 支持（如 NVIDIA 或 OpenCL 库）的 MCMC 代码来制作更快的运行链。
如果有人可以拥有最先进的技术......
MCMC代码优化
计算科学
蒙特卡洛
显卡
随机抽样
样本统计
2021-12-17 20:21:13
3个回答
像Turing.jl这样的 Julia 概率编程库处理抽象数组类型并使用Julia 自动微分工具，因此如果您将模型编写为使用CUDA.jl的东西，那么您的评估将在 GPU 上进行。我已经将它用于大型刚性 PDE 离散化。
像这样的一个？
https://github.com/brandonckelly/CUDAHM
您到底在寻找什么，哪种语言，即插即用方式以及针对哪种问题？
根据情况，pyMC3 或 tensorflow 概率之类的东西可能已经满足您的需求。