More helpful error message if prepare_for_*() not called

Internal report:

If someone is making large changes or writing their own training loop. Users may encounter "illegal memory access" if the model is offloaded and not moved to the GPU before that API call. We should either document or give more helpful error messages when the appropriate prepare step is not called.