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.0
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.0.tar.gz | 5.8 kB | Details |
Built distribution (wheel)
| File | Interpreter | ABI | Platform | Reset |
|---|---|---|---|---|
| vqrotate-0.1.0-py3-none-any.whl | Python 3 | none | any | Details |
Total release size: 12.0 kB
Release files / vqrotate-0.1.0.tar.gz
| Download URL | vqrotate-0.1.0.tar.gz |
|---|---|
| Size | 5.8 kB |
| Tags | Source |
|
SHA-256 checksum How to use checksums |
916422e5efefe8693ec6e554e0a72ba3e570926fe1145c745047a7d440f4da73
|
|
BLAKE2b-256 checksum How to use checksums |
364b6c64bff72f6a2e6f7b9de2b8ff73c9d3ad1105ca2dd29c9eb75e3caedc9e
|
| 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.0-py3-none-any.whl
| Download URL | vqrotate-0.1.0-py3-none-any.whl |
|---|---|
| Size | 6.2 kB |
| Tags | Python 3 |
|
SHA-256 checksum How to use checksums |
bf3091eb532177e98c2518df21d1479f99d887a9f0d3ad587e228c6a73851bca
|
|
BLAKE2b-256 checksum How to use checksums |
04b89e9b09159654cd19dd5c861e5e92a52a3360c32062fce83f8dcd40061f58
|
| Upload date | |
|
Uploaded using Trusted Publishing? What is trusted publishing? |
No |
| Uploaded via |
twine/7.0.0 CPython/3.14.3
|
