from lapjax._src.lax import convolution as convolution
from lapjax._src.lax import qdwh as qdwh
from lapjax._src.lax import svd as svd
from lapjax._src.lax import linalg as linalg
from lapjax._src.lax import slicing as slicing
from lapjax._src.lax import eigh as eigh
from lapjax._src.lax import other as other
from lapjax._src.lax import lax as lax
from lapjax._src.lax import windowed_reductions as windowed_reductions
from lapjax._src.lax import ann as ann
from lapjax._src.lax import utils as utils
from lapjax._src.lax import parallel as parallel
from lapjax._src.lax import stack as stack
from lapjax._src.lax import fft as fft
import sys, importlib
from lapjax.lapsrc.wrapper import _wrap_module
_wrap_module(importlib.import_module(__name__.replace('lapjax', 'jax')), 
             sys.modules[__name__])
