Daniel D. Johnson
Daniel D. Johnson
Hey @ealt, just wanted to ask whether you are still planning to work on this PR? Fixing the HuggingFace conversion seems pretty important. If you don't have the bandwidth to...
Closing this PR since I implemented the changes myself in #125. Hope you don't mind, and thanks for getting this started!
Interesting, thanks for flagging this! This looks like some issue with how JAX handles arrays stored as non-pytree metadata. I was able to reproduce it without using Penzai: ```python import...