Draft: ENH: improve jitting of key functions

This MR adds a new function model_jit_wrapper to safely wrap arbitrary model functions/instances.

I also fixed maybe_jit so that it will work as a wrapper following the version in wcosmo.

This shouldn't change anything in the results, it should just speed up various calculations, but we should run the full set of standard review runs.

Closes #32