Я посмотрел на реализацию вашего JIT и не увидел алгоритмов распределения регистров и т. п.
В основном это просто JIT, выполненный в соответствии с определёнными стандартами.

Может быть, стоит рассмотреть использование готового JIT-движка для этой задачи?