# Sample output object with Nutpie

**URL:** <https://discourse.pymc.io/t/sample-output-object-with-nutpie/16355>\
**Category:** v5\
**Tags:** sampling, nutpie\
**Created:** [January 8, 2025, 6:34pm UTC](https://discourse.pymc.io/t/sample-output-object-with-nutpie/16355 "2025-01-08T18:34:07Z")\
**Posts on this page:** 3\
**Page:** 1

<div class="post-metadata">

**Author:** ![DaveD](https://avatars.discourse-cdn.com/v4/letter/d/bcef8e/32.png) [@DaveD](https://discourse.pymc.io/u/DaveD)\
**Post date:** [January 8, 2025, 6:34pm UTC](https://discourse.pymc.io/t/sample-output-object-with-nutpie/16355/1 "2025-01-08T18:34:07Z")

</div>

Hi,

(Im using PyMC ver 5.19 and Windows 10)

Im using nutpie to compile and sample a model (previously run fine with pm.sample etc). This works very well (_much_ quicker) but the resulting trace object is different (previously I had pm.sample return an InferenceData object).

Is there a straightforward way to create the exact same output object with nutpie.sample, i.e. include the extra attributes (e.g. ‘constant\_data’ & ‘observed\_data’) as in:

Inference data with groups:  
\> posterior  
\> sample\_stats  
\> observed\_data  
\> constant\_data

(Ive tried doing this by hand but its painful).

Thanks

---

<div class="post-metadata">

**Author:** ![ricardoV94](https://yyz2.discourse-cdn.com/flex036/user_avatar/discourse.pymc.io/ricardov94/32/5775_2.png) [@ricardoV94](https://discourse.pymc.io/u/ricardoV94)\
**Post date:** [January 8, 2025, 9:11pm UTC](https://discourse.pymc.io/t/sample-output-object-with-nutpie/16355/2 "2025-01-08T21:11:08Z")

</div>

You can perhaps use these helpers? [pymc/pymc/sampling/jax.py at 82716fbee6d722e4552f88f998e1a20f84045c47 · pymc-devs/pymc · GitHub](https://github.com/pymc-devs/pymc/blob/82716fbee6d722e4552f88f998e1a20f84045c47/pymc/sampling/jax.py#L668-L669)

---

<div class="post-metadata">

**Author:** ![DaveD](https://avatars.discourse-cdn.com/v4/letter/d/bcef8e/32.png) [@DaveD](https://discourse.pymc.io/u/DaveD)\
**Post date:** [January 9, 2025, 1:45pm UTC](https://discourse.pymc.io/t/sample-output-object-with-nutpie/16355/3 "2025-01-09T13:45:14Z")

</div>

Works perfectly, thank you