mrku69

mamba

Mamba SSM architecture

mamba-tiny

Simple, minimal implementation of the Mamba SSM in one pytorch file. More efficient than using for loops, but probably less efficient than using associative scans

llm-viz

3D Visualization of an GPT-style LLM

narrator

David Attenborough narrates your life

simplified_transformers

H-Deformable-DETR

[CVPR2023] This is an official implementation of paper "DETRs with Hybrid Matching".

U-ViT

A PyTorch implementation of the paper "All are Worth Words: A ViT Backbone for Diffusion Models".

denoising-diffusion-pytorch

Implementation of Denoising Diffusion Probabilistic Model in Pytorch

Revisiting-Dis-vs-Gen-Classifiers

Official implementation for "Revisiting Discriminative vs. Generative Classifiers: Theory and Implications".

HAWQ

Quantization library for PyTorch. Support low-precision and mixed-precision quantization, with hardware implementation through TVM.

DiffusionDet

[ICCV2023] PyTorch implementation of DiffusionDet (https://arxiv.org/abs/2211.09788)

tutorials

PyTorch tutorials.

detrex

detrex is a research platform for Transformer-based Object Detection algorithms including DETR (ECCV 2020), Deformable-DETR (ICLR 2021), Conditional-DETR (ICCV 2021), DAB-DETR (ICLR 2022), DN-DETR (CVPR 2022), DINO (arXiv 2022), etc.

vidt

FQ-ViT

[IJCAI 2022] FQ-ViT: Post-Training Quantization for Fully Quantized Vision Transformer

T2T-ViT

ICCV2021, Tokens-to-Token ViT: Training Vision Transformers from Scratch on ImageNet

ViTDet

Unofficial implementation of Exploring Plain Vision Transformer Backbones for Object Detection

transformer-from-scratch

Well documented, unit tested, type checked and formatted implementation of a vanilla transformer - for educational purposes.

vit-explain

Explainability for Vision Transformers

mobilevit-pytorch

A PyTorch implementation of "MobileViT: Light-weight, General-purpose, and Mobile-friendly Vision Transformer".

the-clean-transformer

pytorch-lightning과 wandb로 깔끔하게 구현해보는 트랜스포머

ssd.pytorch

A PyTorch Implementation of Single Shot MultiBox Detector

Transformer-in-Vision

Recent Transformer-based CV and related works.

attention-is-all-you-need-pytorch

A PyTorch implementation of the Transformer model in "Attention is All You Need".

pytorch-image-models

PyTorch image models, scripts, pretrained weights -- ResNet, ResNeXT, EfficientNet, EfficientNetV2, NFNet, Vision Transformer, MixNet, MobileNet-V3/V2, RegNet, DPN, CSPNet, and more

Swin-Transformer

This is an official implementation for "Swin Transformer: Hierarchical Vision Transformer using Shifted Windows".

Deformable-DETR

Deformable DETR: Deformable Transformers for End-to-End Object Detection.

up-detr

[CVPR2021 Oral] UP-DETR: Unsupervised Pre-training for Object Detection with Transformers

RelationNet2

RelationNet++: Bridging Visual Representations via Attention Mechanism for Object Detection

overleaf

A web-based collaborative LaTeX editor