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.2
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.2.tar.gz | 5.7 kB | Details |
Built distribution (wheel)
| File | Interpreter | ABI | Platform | Reset |
|---|---|---|---|---|
| vqrotate-0.1.2-py3-none-any.whl | Python 3 | none | any | Details |
Total release size: 12.0 kB
Release files / vqrotate-0.1.2.tar.gz
| Download URL | vqrotate-0.1.2.tar.gz |
|---|---|
| Size | 5.7 kB |
| Tags | Source |
|
SHA-256 checksum How to use checksums |
650fa2cfbe16f747651e4143e4adf8abd51e02b1f66b2d5b90a5e837950d7dea
|
|
BLAKE2b-256 checksum How to use checksums |
041f3bf494d10b5d48c174689c0e2ce54218c60499354a951c7dd890d8cafb74
|
| 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.2-py3-none-any.whl
| Download URL | vqrotate-0.1.2-py3-none-any.whl |
|---|---|
| Size | 6.3 kB |
| Tags | Python 3 |
|
SHA-256 checksum How to use checksums |
07e9fc9d5a771c9c60a1f56f6347e3c60a9ec4889c92e3dd8071f4020d8e76f2
|
|
BLAKE2b-256 checksum How to use checksums |
05b2f3aa18ac06c35ac4b63ce359b6c42080a45c3b41505de58e7cbde07cc93d
|
| Upload date | |
|
Uploaded using Trusted Publishing? What is trusted publishing? |
No |
| Uploaded via |
twine/7.0.0 CPython/3.14.3
|
