from lapjax.scipy.stats import norm as norm
from lapjax.scipy.stats import t as t
from lapjax.scipy.stats import logistic as logistic
from lapjax.scipy.stats import truncnorm as truncnorm
from lapjax.scipy.stats import betabinom as betabinom
from lapjax.scipy.stats import gennorm as gennorm
from lapjax.scipy.stats import multinomial as multinomial
from lapjax.scipy.stats import beta as beta
from lapjax.scipy.stats import poisson as poisson
from lapjax.scipy.stats import expon as expon
from lapjax.scipy.stats import cauchy as cauchy
from lapjax.scipy.stats import geom as geom
from lapjax.scipy.stats import pareto as pareto
from lapjax.scipy.stats import uniform as uniform
from lapjax.scipy.stats import multivariate_normal as multivariate_normal
from lapjax.scipy.stats import laplace as laplace
from lapjax.scipy.stats import bernoulli as bernoulli
from lapjax.scipy.stats import gamma as gamma
from lapjax.scipy.stats import dirichlet as dirichlet
from lapjax.scipy.stats import chi2 as chi2
from lapjax.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__])
