vqrotate
One-line Rotation Trick for VQ-VAEs.
Drop-in wrapper that attaches the ICLR 2025 Rotation Trick to any VQ layer.
Features
- Single line of code:
attach_rotation(quantizer) - Multi-head support (lucidrains compatible)
- Mixed precision (AMP) ready
- Adaptive gradient scaling (89% codebook usage on Speech Commands)
Install
pip install git+https://github.com/develepo/vqrotate.git
1 maintainer
Release files for vqrotate 0.1.1
For a detailed explanation of source distributions (sdists) and built distributions (wheels), please see the package formats documentation.
Source distribution (sdist)
| File | Size | Uploaded | |
|---|---|---|---|
| vqrotate-0.1.1.tar.gz | 5.9 kB | Details |
Built distribution (wheel)
| File | Interpreter | ABI | Platform | Reset |
|---|---|---|---|---|
| vqrotate-0.1.1-py3-none-any.whl | Python 3 | none | any | Details |
Total release size: 12.1 kB
Release files / vqrotate-0.1.1.tar.gz
| Download URL | vqrotate-0.1.1.tar.gz |
|---|---|
| Size | 5.9 kB |
| Tags | Source |
|
SHA-256 checksum How to use checksums |
f0cf56bc881119ec56a688a37270d7d8f8e67a5889974b7263d499ea627f23a0
|
|
BLAKE2b-256 checksum How to use checksums |
e392d519ad9e43dcd39239f3f35cf3695a955e5de71459ad5de965af3728a12d
|
| Upload date | |
|
Uploaded using Trusted Publishing? What is trusted publishing? |
No |
| Uploaded via |
twine/7.0.0 CPython/3.14.3
|
Release files / vqrotate-0.1.1-py3-none-any.whl
| Download URL | vqrotate-0.1.1-py3-none-any.whl |
|---|---|
| Size | 6.3 kB |
| Tags | Python 3 |
|
SHA-256 checksum How to use checksums |
fe3a0e2913366e704287acbc164c2f15aa11ef7f7ae020b90e7c867b51ad916c
|
|
BLAKE2b-256 checksum How to use checksums |
07ede3182fa8222d3b095043bcc3af3568e1993c2d55c3bc887fc774cbfa02f4
|
| Upload date | |
|
Uploaded using Trusted Publishing? What is trusted publishing? |
No |
| Uploaded via |
twine/7.0.0 CPython/3.14.3
|
