CUQI-DTU / CUQIpy

https://cuqi-dtu.github.io/CUQIpy/
Apache License 2.0
44 stars 9 forks source link

test NUTS checkpointing/state/history #404

Closed amal-ghamdi closed 3 weeks ago

amal-ghamdi commented 6 months ago

Description:

NUTS does not pass history, state and checkpointing tests. We want to fix this.

DoD:

Original issue:

Furthermore, consider getting rid of the _pre_warmup and _pre_sample methods. See Discussion in #408

amal-ghamdi commented 1 month ago

@jakobsj , can you check the DoD, please? Thanks!

jakobsj commented 3 weeks ago

@amal-ghamdi DoD approved!