from lapjax.scipy import linalg as linalg
from lapjax.scipy import signal as signal
from lapjax.scipy import special as special
from lapjax.scipy import fft as fft
from lapjax.scipy import ndimage as ndimage
import sys, importlib
from lapjax.lapsrc.wrapper import _wrap_module
_wrap_module(importlib.import_module(__name__.replace('lapjax', 'jax')), 
             sys.modules[__name__])
