Recherche de support GPU sur Markov-chain Monte-Carlo (MCMC) en Python - méthode Hamiltonian MCMC

Solution du problème

Je ne connais pas la bibliothèque que vous utilisez, mais je ne vois pas comment y ajouter le support GPU. Cependant, pour votre problème spécifique, je recommanderais une bibliothèque appelée Pyro. Il est construit sur PyTorch, il prend donc en charge le GPU. Voici un exemple d'utilisation de MCMC dans Pyro : https://pyro.ai/examples/mcmc.html