Initializing a model instance before loading

You could use some utility funcitions from this topic or load your jitted model directly (which seems to be the recommended way).