Is there a way to save and load an ArrayBox to hdf5 file while retaining derivative information?
I took an approach of trying to save each of the _value, _trace, and _node, but the VJPNode seems to contain function information and I could not serialize it.
Any insight is appreciated!