[schmidtjonathan] Pickled model, training state, and config

initial commit