Hello, I tried implementing PyMC v4.0 with Jax and GPU. Using pm.sampling_jax.sample_numpyro_nuts, it works / samples fine and provides a similar posterior to PyMC3; however, it does not return the ‘observed_data’ object within InferenceData. I noticed in the release notes that ‘inner workings ha…

PyMC v4 not returning 'observed_data' in InferenceData object

zaxtax March 18, 2022, 11:54am 6

Yep, I can reproduce this error. The issue has been re-opened so you can follow the progress there.

Topic		Replies	Views
Need help for trace in jax_sampling Questions	12	795	November 7, 2021
How to define reasonable observed data Questions	7	741	December 14, 2017
Confusion between latest release version and development version of PyMC Questions	7	544	October 31, 2021
Bernoulli Distribution issues when using Data class Questions	4	670	March 5, 2021
My misunderstanding with the argument 'observed' Questions	2	404	March 6, 2020