static_cg: JIT `mat` to reduce compile time

This shaves off about 20% of the compile time of the static_cg in the current nifty_re demo.

This is safe because the while_loop statement within static_cg would have compiled everything anyways.