Open
Description
Description
This feature request stems from this discussion. The idea is to add the ability to record and store the step methods' sampling_state
to the NDArray
backend. Potentially, also to also record it once sampling finishes.
The ZarrTrace
backend development that gave birth to this idea used a link_stepper
method to gain a reference to the step method and then get its sampling_state
. This comment suggested adding the method to NDArray
as well.