from lapjax.experimental.compilation_cache import gfile_cache as gfile_cache
from lapjax.experimental.compilation_cache import compilation_cache as compilation_cache
from lapjax.experimental.compilation_cache import cache_interface as cache_interface
import sys, importlib
from lapjax.lapsrc.wrapper import _wrap_module
_wrap_module(importlib.import_module(__name__.replace('lapjax', 'jax')), 
             sys.modules[__name__])
