Stuck on an issue?

Lightrun Answers was designed to reduce the constant googling that comes with debugging 3rd party libraries. It collects links to all the places you might be looking at while hunting down a tough bug.

And, if you’re still stuck at the end, we’re happy to hop on a call to see how we can help out.

Add some kind de/serialisation?

See original GitHub issue

Equinox models are just PyTrees so they should be very easy to serialise/deserialise; just save the PyTree to disk in whatever way is desired. It might be worth adding some library functions for this just for convenience. Perhaps checking the device of JAX arrays etc?

This should respect the get_state/set_state stuff that’s being put together.

In addition, there should be a version of get_state which inlines its state in the call graph, for faster inference.

Issue Analytics

State:
Created a year ago
Comments:13 (8 by maintainers)

Top GitHub Comments

1reaction

jaschaucommented, May 4, 2022

Hi @patrick-kidger, sorry for the little progress here. I’ve been side-tracked with other topics so I haven’t come around to working on this beyond our initial discussion. I’m afraid I cannot promise any serious progress on this in the upcoming weeks from my side, so please move ahead if you’re eager to work on this.

0reactions

patrick-kidgercommented, May 4, 2022

@jaschau Any plans to pick this up as a PR? No worries if not – I’ll do it – I just want to get seralisation+deserialisation into the next release of Equinox.