Support custom memory allocators in the JIT compiler.