alphaRGB

performer-pytorch

An implementation of Performer, a linear attention-based transformer, in Pytorch

vision-longformer

Evo-ViT

Official implement of Evo-ViT: Slow-Fast Token Evolution for Dynamic Vision Transformer

SPViT

[TPAMI 2024] This is the official repository for our paper: ''Pruning Self-attentions into Convolutional Layers in Single Path''.

nn_pruning

Prune a model while finetuning or training.

lite-transformer

[ICLR 2020] Lite Transformer with Long-Short Range Attention

retraining-free-pruning

[NeurIPS 2022] A Fast Post-Training Pruning Framework for Transformers

flash-attention

Fast and memory-efficient exact attention

KAIR

Image Restoration Toolbox (PyTorch). Training and testing codes for DPIR, USRNet, DnCNN, FFDNet, SRMD, DPSR, BSRGAN, SwinIR

Awesome-Visual-Transformer

Collect some papers about transformer with vision. Awesome Transformer with Computer Vision (CV)

self-attention-cv

Implementation of various self-attention mechanisms focused on computer vision. Ongoing repository.

ml-visuals

🎨 ML Visuals contains figures and templates which you can reuse and customize to improve your scientific writing.

AIChip_Paper_List

cuSPARSE

nlp_paper_study

该仓库主要记录 NLP 算法工程师相关的顶会论文研读笔记

recurrent-transformer

[ACL 2020] PyTorch code for MART: Memory-Augmented Recurrent Transformer for Coherent Video Paragraph Captioning

dash-detr

A User Interface for DETR built with Dash. 100% Python.

pytorch-image-models

The largest collection of PyTorch image encoders / backbones. Including train, eval, inference, export scripts, and pretrained weights -- ResNet, ResNeXT, EfficientNet, NFNet, Vision Transformer (ViT), MobileNetV4, MobileNet-V3 & V2, RegNet, DPN, CSPNet, Swin Transformer, MaxViT, CoAtNet, ConvNeXt, and more

vit-pytorch

Implementation of Vision Transformer, a simple way to achieve SOTA in vision classification with only a single transformer encoder, in Pytorch

cbc-ubuntu-64bit

Cb (C flat) compiler. Cb is simplified C.

DeepSpeed

DeepSpeed is a deep learning optimization library that makes distributed training and inference easy, efficient, and effective.

Awesome-Vision-Transformer-Collection

Variants of Vision Transformer and its downstream tasks

cpc

建党 100 周年！写一个名为 CPC 的 C 编译器，为党庆生！

cuda-cuckoo-hashing

ShanghaiTech CS121 Parallel Computing Fall 2021 Lab2: Cuckoo Hashing using CUDA

CuckooHashing

Concurrent cuckoo hash table implemented in CUDA

parallelHash

Parallel Hashing using Cuckoo's Algorithm for GPU using CUDA.

cuckoo

cuckoo

a memory-bound graph-theoretic proof-of-work system

hash-table

Fast, reliable cuckoo hash table for Node.js.

pymtcnn_facedetector

MTCNN Face detector implemented by C++ and Python