test: cover compile behavior; respect torch_compile kwarg in load_checkpoint

Add tests verifying model.forward is compiled when torch_compile=True and
left as a plain bound method when False.

Also let an explicit load_checkpoint(path, torch_compile=...) override the
instance default instead of being silently ignored, and pop the kwarg
before forwarding to the inner module (which does not use it).