from .stochastic import GaussianSample, GumbelSoftmax, BaseSample, Sample, NormalSample, LaplaceSample, GaussianMerge
from .stochastic_types import ModuleXToXTupleYi, XTupleYi

from .stochastic import GaussianSample, GumbelSoftmax, BaseSample, Sample, NormalSample, LaplaceSample, GaussianMerge
from .stochastic_types import ModuleXToXTupleYi, XTupleYi
