Vector Quantization - Pytorch
12 similar packages · ranked by shared classifiers & health score · View vector-quantize-pytorch →
Image augmentation library for deep neural networks
Conditionally Routed Attention
Easily turn a set of image urls to an image dataset
BS-RoFormer - Band-Split Rotary Transformer for SOTA Music Source Separation
BS-RoFormer - Band-Split Rotary Transformer for SOTA Music Source Separation
Easy way to keep track of exponential moving average version of your pytorch module
Local attention, window with lookback, for language modeling
Product Key Memory
Einops Extensions
Resize Right
The convolutional module from the Conformer paper
Rotary Embedding - Pytorch