Mirage: Automatically Generating Fast GPU Kernels without Programming in Triton/CUDA

Implemetation of ELIC: Efficient Learned Image Compression with Unevenly Grouped Space-Channel Contextual Adaptive Coding

[ICLR2024] FTIC: Frequency-aware Transformer for Learned Image Compression

PyTorch implementation of MAR+DiffLoss https://arxiv.org/abs/2406.11838

Vision-RWKV: Efficient and Scalable Visual Perception with RWKV-Like Architectures

[ECCV 2024] The official code of paper "Open-Vocabulary SAM".

PyTorch implementation of PerCo (Towards Image Compression with Perfect Realism at Ultra-Low Bitrates, ICLR 2024)

Official PyTorch Implementation of MambaVision: A Hybrid Mamba-Transformer Vision Backbone

[EMNLP 2023 Main] Sparse Low-rank Adaptation of Pre-trained Language Models

👁️ 🖼️ 🔥PyTorch Toolbox for Image Quality Assessment, including PSNR, SSIM, LPIPS, FID, NIQE, NRQM(Ma), MUSIQ, TOPIQ, NIMA, DBCNN, BRISQUE, PI and more...

PyTorch Implementation of the CVPR'24 Paper "Learned Lossless Image Compression based on Bit Plane Slicing"

Efficient implementation of Depthwise Conv2d. The main code is from https://github.com/MegEngine

Pix2Seq codebase: multi-tasks with generative modeling (autoregressive and diffusion)

New generation entropy codecs : Finite State Entropy and Huff0

Official PyTorch Implementation of "Scalable Diffusion Models with Transformers"

Official code of Remote Sensing Mamba

Neighborhood Attention Extension. Bringing attention to a neighborhood near you!

RWKV (pronounced RwaKuv) is an RNN with great LLM performance, which can also be directly trained like a GPT transformer (parallelizable). We are at RWKV-7 "Goose". So it's combining the best of RN…

Code for paper LocalMamba: Visual State Space Model with Windowed Selective Scan

Code Implementation of EfficientVMamba

[ICML 2024] Vision Mamba: Efficient Visual Representation Learning with Bidirectional State Space Model

VMamba: Visual State Space Models，code is based on mamba

Deep Lossy Plus Residual Coding for Lossless and Near-lossless Image Compression (TPAMI'24, Journal Extension of our CVPR'21 work)

关于视频编解码学习资料

Official Pytorch Implementation of Our CVPR2023 Paper: "Towards Accurate Image Coding: Improved Autoregressive Image Generation with Dynamic Vector Quantization"

PhotoMaker [CVPR 2024]

Official Implementation for (ICML 2023) Bit Allocation using Optimization