VectorQuantizationΒΆ

Implements vector quantization (VQ)/online k-means clustering.

Abstract Signature:

VectorQuantization(num_clusters: int = 0, num_heads: int = 0, dim_per_head: int = 0)

PyTorch

API: β€”
Strategy: Custom / Partial

PaxML / Praxis

API: praxis.layers.VectorQuantization
Strategy: Direct Mapping