Rupesh K Srivastava's starred repositories
stable-diffusion
A latent text-to-image diffusion model
vit-pytorch
Implementation of Vision Transformer, a simple way to achieve SOTA in vision classification with only a single transformer encoder, in Pytorch
FasterTransformer
Transformer related optimization, including BERT, GPT
AITemplate
AITemplate is a Python framework which renders neural network into high performance CUDA/HIP C++ code. Specialized for FP16 TensorCore (NVIDIA GPU) and MatrixCore (AMD GPU) inference.
x-transformers
A simple but complete full-attention transformer with a set of promising experimental features from various papers
k-diffusion
Karras et al. (2022) diffusion models for PyTorch
transformer-deploy
Efficient, scalable and enterprise-grade CPU/GPU inference server for 🤗 Hugging Face transformer models 🚀
WeightWatcher
The WeightWatcher tool for predicting the accuracy of Deep Neural Networks
Versatile-Diffusion
Versatile Diffusion: Text, Images and Variations All in One Diffusion Model, arXiv 2022 / ICCV 2023
lovely-tensors
Tensors, ready for human consumption
Neighborhood-Attention-Transformer
Neighborhood Attention Transformer, arxiv 2022 / CVPR 2023. Dilated Neighborhood Attention Transformer, arxiv 2022
TorchSnooper
Debug PyTorch code using PySnooper
pylustrator
Visualisations of data are at the core of every publication of scientific research results. They have to be as clear as possible to facilitate the communication of research. As data can have different formats and shapes, the visualisations often have to be adapted to reflect the data as well as possible. We developed Pylustrator, an interface to directly edit python generated matplotlib graphs to finalize them for publication. Therefore, subplots can be resized and dragged around by the mouse, text and annotations can be added. The changes can be saved to the initial plot file as python code.
python-on-whales
An awesome Python wrapper for an awesome Docker CLI!
imagenet-sample-images
1000 images, one per image-net class. For easy visualization/exploration of classes.
gpu-tester
gpu tester detects broken and slow gpus in a cluster
pytorch-symbolic
Provides symbolic API for model creation in PyTorch.