pranavphoenix / WaveMix Star 91 Code Issues Pull requests 2D discrete Wavelet Transform for Image Classification and Segmentation efficient linear image-classification attention semantic-segmentation wavelet-transform token-mixing Updated Jan 6, 2025 Python
lucidrains / metaformer-gpt Star 26 Code Issues Pull requests Implementation of Metaformer, but in an autoregressive manner deep-learning artificial-intelligence autoregressive token-mixing Updated Jun 21, 2022 Python
pranavphoenix / WavePaint Star 26 Code Issues Pull requests Wavelet-based Image InPainting Model deep-neural-networks computer-vision cnn pytorch image-generation image-inpainting inpainting wavelet-transform inpainting-methods inpainting-algorithm token-mixing Updated Sep 15, 2024 Python
Yang916-yy / LSSO Star 3 Code Issues Pull requests A fast, stable alternative to attention: closed-form low-rank token mixing for long-context PyTorch models, with CUDA kernels. computer-vision deep-learning genomics cuda transformers pytorch attention low-rank sequence-modeling vision-transformers efficient-transformers long-context token-mixing structured-optimization accretive-equilibrium long-range-modeling Updated Aug 1, 2026 Python