Fix saving and loading

[kulbachcedric]

Zoo models must fully restore their trained weights, dynamically expanded architecture, optimizer states, and runtime data.

Make also tests for this to ensure the behavior.