r/Python•
Tapaiz

Let's discuss: vqrotate — a PyPI library for the ICLR 2025 Rotation Trick (VQ‑VAEs)

Discussion

Let's discuss on it :

A PyTorch library that wraps any VQ layer and applies the Rotation Trick in one line:

```python

from vqrotate import attach_rotator

attach_rotator(quantizer, strategy="rotation")

```

Validated on Speech Commands (audio):

STE: 19.99% codebook usage

Rotation: 34.18% (+14.19 pts)

Matrix‑free Householder reflection with O(d) memory.

Git hub: develepo - vqrotate

PyPI: pip install vqrotate

Docs include derivations, API reference, and benchmark results. Feedback welcome.

Posted Aug 13, 2026, 5:00 PM UTC

00
Loading comments…