Я реализую иерархическую фармакокинетическую (PK) модель в PyMC, используя Diffrax для решения ОДУ через оболочку icomo.jax2pytensor. Модель успешно компилируется и инициализируется, но Выборка MCMC зависает на неопределенный срок при прогрессе 0% как с сэмплерами NumPyro, так и с BlackJAX.
Код: Выделить всё
Running chain 0: 0%| | 0/2000 [03:51
Подробнее здесь: [url]https://stackoverflow.com/questions/79863500/pymc-diffrax-mcmc-sampler-hangs-at-0-for-hierarchical-pk-ode-model-with-jax[/url]