The following was part of a larger patch, but I've pruned the patch to illustrate the issue--- that the output of [jit.matrix] is expected to be 16x16 from the very beginning, but isn't. See the red comment in the attached patch.