Discussion : I built a PyTorch library that wraps any VQ layer and applies the Rotation Trick in one