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