from .quantizer import VectorQuantizer, DiagonalGaussianDistribution
from .mvq import VectorQuantizerMVQ
from .softvq import SoftVectorQuantizer