Yaning Liang

Hi, sorry about the delay. The issue is now fixed. Can you please try again? Thanks!

Hi Chris, Thanks for raising the issue. We have submitted a fix and this issue should have been resolved now. Can you please verify that this is no longer erroring...

Hi @chiamp, thanks for raising the issue. We had to update the pytree key names encoded since special characters like '~' were not encoded properly with the previous encoding. The...

Hi Chris, Do you mind checking what version of the orbax checkpointing you are using? I just checked this in notebook with v0.4.1 and it works.