from .functions import QuantizationFunction, LearnableQuantization, FixedQuantization, VectorQuantization, \
    LearnableVectorQuantization, learnable_vector_quantization, trained_vector_quantization
from .vqvae import VQVAE, vqvae
