Easy loader

Goal: make model eval more ergonomic and pre-jit.

• alternate version of load_trained_model, jitting in the params
• test fns