from lapjax._src.scipy.stats import norm as norm
from lapjax._src.scipy.stats import t as t
from lapjax._src.scipy.stats import logistic as logistic
from lapjax._src.scipy.stats import truncnorm as truncnorm
from lapjax._src.scipy.stats import betabinom as betabinom
from lapjax._src.scipy.stats import gennorm as gennorm
from lapjax._src.scipy.stats import multinomial as multinomial
from lapjax._src.scipy.stats import beta as beta
from lapjax._src.scipy.stats import poisson as poisson
from lapjax._src.scipy.stats import expon as expon
from lapjax._src.scipy.stats import cauchy as cauchy
from lapjax._src.scipy.stats import geom as geom
from lapjax._src.scipy.stats import pareto as pareto
from lapjax._src.scipy.stats import uniform as uniform
from lapjax._src.scipy.stats import multivariate_normal as multivariate_normal
from lapjax._src.scipy.stats import laplace as laplace
from lapjax._src.scipy.stats import bernoulli as bernoulli
from lapjax._src.scipy.stats import gamma as gamma
from lapjax._src.scipy.stats import _core as _core
from lapjax._src.scipy.stats import dirichlet as dirichlet
from lapjax._src.scipy.stats import kde as kde
from lapjax._src.scipy.stats import chi2 as chi2
from lapjax._src.scipy.stats import nbinom as nbinom
import sys, importlib
from lapjax.lapsrc.wrapper import _wrap_module
_wrap_module(importlib.import_module(__name__.replace('lapjax', 'jax')), 
             sys.modules[__name__])
