Wonjae Kim's repositories
amdim-public
Public repo for Augmented Multiscale Deep InfoMax representation learning
dandelin.github.io
https://wonjae.kim
deit
Official DeiT repository
detr
End-to-End Object Detection with Transformers
dwarfs
A fast high compression read-only file system
image-to-set
The project is about predicting sets (of classes) from images.
minGPT
A minimal PyTorch re-implementation of the OpenAI GPT (Generative Pretrained Transformer) training
PaintTransformer
PyTorch implementation of paper: Paint Transformer: Feed Forward Neural Painting with Stroke Prediction, ICCV 2021.
PyPDF2
A utility to read and write PDFs with Python
pytorch-image-models
PyTorch image models, scripts, pretrained weights -- (SE)ResNet/ResNeXT, DPN, EfficientNet, MixNet, MobileNet-V3/V2, MNASNet, Single-Path NAS, FBNet, and more
pytorch-lightning
The lightweight PyTorch wrapper for high-performance AI research. Scale your models, not the boilerplate.
pytorch-lightning-bolts
Toolbox of models, callbacks, and datasets for AI/ML researchers.
reformer-pytorch
Reformer, the efficient Transformer, in Pytorch
Swin-Transformer
This is an official implementation for "Swin Transformer: Hierarchical Vision Transformer using Shifted Windows".
torchdyn
A PyTorch based library for all things neural differential equations
transformers
🤗Transformers: State-of-the-art Natural Language Processing for Pytorch and TensorFlow 2.0.
UNITER
Research code for ECCV 2020 paper "UNITER: UNiversal Image-TExt Representation Learning"
vision
Datasets, Transforms and Models specific to Computer Vision
ViT-pytorch
Pytorch reimplementation of the Vision Transformer (An Image is Worth 16x16 Words: Transformers for Image Recognition at Scale)
YOLOS
You Only Look at One Sequence (https://arxiv.org/abs/2106.00666)