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