from lapjax._src.numpy import reductions as reductions
from lapjax._src.numpy import ndarray as ndarray
from lapjax._src.numpy import linalg as linalg
from lapjax._src.numpy import lax_numpy as lax_numpy
from lapjax._src.numpy import index_tricks as index_tricks
from lapjax._src.numpy import vectorize as vectorize
from lapjax._src.numpy import polynomial as polynomial
from lapjax._src.numpy import ufuncs as ufuncs
from lapjax._src.numpy import setops as setops
from lapjax._src.numpy import util as util
from lapjax._src.numpy 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__])
