from lapjax._src.scipy import linalg as linalg
from lapjax._src.scipy import signal as signal
from lapjax._src.scipy import special as special
from lapjax._src.scipy import fft as fft
from lapjax._src.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__])
