Taming Transformers for High-Resolution Image Synthesis

Official codebase used to develop Vision Transformer, SigLIP, MLP-Mixer, LiT and more.

A python tutorial on bayesian modeling techniques (PyMC3)

Easily compute clip embeddings and build a clip retrieval system with them

[CVPR 2021] Official PyTorch implementation for Transformer Interpretability Beyond Attention Visualization, a novel method to visualize classifications by Transformer based networks.

A framework for data augmentation for 2D and 3D image classification and segmentation

Course material for STAT 479: Machine Learning (FS 2018) at University Wisconsin-Madison

Code for my medium article

Python machine learning applications in image processing, recommender system, matrix completion, netflix problem and algorithm implementations including Co-clustering, Funk SVD, SVD++, Non-negative…

Detect model's attention

Deep Feature Factorization

This code implements sparse coding in PyTorch with GPU support.

Triplet Loss for Knowledge Distillation

This is a simple Chainer implementation of the Wasserstein GAN on MNIST digits.

Implementation of Projected gradient NMF

Neural NMF