mcx icon indicating copy to clipboard operation
mcx copied to clipboard
verified publisher icon rlouf

Save the kernel parameters after warmup

Open rlouf opened this issue 5 years ago • 0 comments

https://github.com/rlouf/mcx/blob/2a2b94801e68d94d86826863eeee80f0b84c390d/mcx/inference/hmc.py#L198-L205

We currently pass the parameters directly to the runtime; While the values of the parameters are passed in the Trace object, it would be convenient to update the Kernel's parameter values.

rlouf avatar Mar 10 '21 15:03 rlouf