Prajwal Singh's starred repositories
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
albumentations
Fast and flexible image augmentation library. Paper about the library: https://www.mdpi.com/2078-2489/11/2/125
stylegan2-ada-pytorch
StyleGAN2-ADA - Official PyTorch implementation
tiny-cuda-nn
Lightning fast C++/CUDA neural network framework
DirectVoxGO
Direct voxel grid optimization for fast radiance field reconstruction.
OpenAI-CLIP
Simple implementation of OpenAI CLIP model in PyTorch.
EEG-Conformer
EEG Transformer 2.0. i. Convolutional Transformer for EEG Decoding. ii. Novel visualization - Class Activation Topography.
taichi-ngp-renderer
An Instants-NGP renderer that has been implemented using Taichi
torch-stft
An STFT/iSTFT for PyTorch.
batchrenorm
Batch Renormalization in Pytorch