__init__.py

from pytorch_widedeep.bayesian_models import bayesian_nn
from pytorch_widedeep.bayesian_models.tabular import (
    BayesianWide,
    BayesianTabMlp,
)