Sereiwathna's starred repositories
flash-attention
Fast and memory-efficient exact attention
transformers
🤗 Transformers: State-of-the-art Machine Learning for Pytorch, TensorFlow, and JAX.
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), MobileNet-V3/V2, RegNet, DPN, CSPNet, Swin Transformer, MaxViT, CoAtNet, ConvNeXt, and more
vision-transformers-cifar10
Let's train vision transformers (ViT) for cifar 10!
pytorch-cifar100
Practice on cifar100(ResNet, DenseNet, VGG, GoogleNet, InceptionV3, InceptionV4, Inception-ResNetv2, Xception, Resnet In Resnet, ResNext,ShuffleNet, ShuffleNetv2, MobileNet, MobileNetv2, SqueezeNet, NasNet, Residual Attention Network, SENet, WideResNet)
Vision-Transformer-CIFAR10
Pytorch implementation of some vision transformers, trained on CIFAR-10.
Awesome-Transformer-Attention
An ultimately comprehensive paper list of Vision Transformer/Attention, including papers, codes, and related websites
a-PyTorch-Tutorial-to-Image-Captioning
Show, Attend, and Tell | a PyTorch Tutorial to Image Captioning
Deep-Tutorials-for-PyTorch
In-depth tutorials for implementing deep learning models on your own with PyTorch.
pytorch-cifar-models
Pretrained models on CIFAR10/100 in PyTorch
ViT-pytorch
Pytorch reimplementation of the Vision Transformer (An Image is Worth 16x16 Words: Transformers for Image Recognition at Scale)
Deep-Learning-Papers-Reading-Roadmap
Deep Learning papers reading roadmap for anyone who are eager to learn this amazing tech!
Poisson_flow
Code for NeurIPS 2022 Paper, "Poisson Flow Generative Models" (PFGM)
faster-rcnn.pytorch
A faster pytorch implementation of faster r-cnn
deep_learning_object_detection
A paper list of object detection using deep learning.
lecture_5_yolov5
YOLOv5 🚀 in PyTorch > ONNX > CoreML > TFLite
awesome-machine-learning
A curated list of awesome Machine Learning frameworks, libraries and software.
easy-fpn.pytorch
An easy implementation of FPN (https://arxiv.org/pdf/1612.03144.pdf) in PyTorch.
the-gan-zoo
A list of all named GANs!
pytorch-multigpu
Multi GPU Training Code for Deep Learning with PyTorch
pytorch-pruning
PyTorch Implementation of [1611.06440] Pruning Convolutional Neural Networks for Resource Efficient Inference
ML-From-Scratch
Machine Learning From Scratch. Bare bones NumPy implementations of machine learning models and algorithms with a focus on accessibility. Aims to cover everything from linear regression to deep learning.
PyTorch-YOLOv3
Minimal PyTorch implementation of YOLOv3
riemannian-score-sde
Score-based generative models for compact manifolds
Image-Adaptive-3DLUT
Learning Image-adaptive 3D Lookup Tables for High Performance Photo Enhancement in Real-time
PyTorch-GAN
PyTorch implementations of Generative Adversarial Networks.