I am looking for MCMC codes with a GPU suport (like NVIDIA or OpenCL libraries) to make faster run chains.
If someone could have a state of the art ...
Computational Science Stack Exchange is a question and answer site for scientists using computers to solve scientific problems. It only takes a minute to sign up.
Sign up to join this communityI am looking for MCMC codes with a GPU suport (like NVIDIA or OpenCL libraries) to make faster run chains.
If someone could have a state of the art ...
The Julia probabilistic programming libraries like Turing.jl work on abstract array types and use the Julia automatic differentiation tooling, so if you write your model as something that uses CUDA.jl then your evaluations will take place on the GPU. I've used this for large stiff PDE discretizations.
One like this one?
https://github.com/brandonckelly/CUDAHM
What exactly are you looking for, which language, how plug&play and for which kind of problem?
Depending on the case stuff like pyMC3 or tensorflow probability might already suit your needs.
The library that pymc uses to do this is aesara.
https://github.com/aesara-devs/aesara
Stan uses OpenCL