equinox icon indicating copy to clipboard operation
equinox copied to clipboard

Published 20 hours ago • verified publisher icon patrick-kidger

Orbax Integration

Open allen-adastra opened this issue 7 months ago • 4 comments

Integrating with Orbax would be great. I actually have a working class EquinoxCheckPointHandler(ocp.CheckpointHandler) that seems to work for generic instances of eqx.Module. Would there be interest in finding a way to bring that in to Equinox officially?

allen-adastra avatar Nov 29 '23 10:11 allen-adastra