Implement JIT compilation