Aravind Srinivas

flowpp. Code for reproducing Flow ++ experiments

neural-mpc. Python

curl_rainbow. Python

upn. Python

minGPT. A minimal PyTorch re-implementation of the OpenAI GPT (Generative Pretrained Transformer) training

pycls. Codebase for Image Classification Research, written in PyTorch.

awesome-critical-tech-reading-list. A reading list for the modern critical programmer

ML-Systems. papers on scalable and efficient machine learning systems

upn-icml-stuff.

simclr. SimCLRv2 - Big Self-Supervised Models are Strong Semi-Supervised Learners

stable-diffusion. Jupyter Notebook

CompilerGym. A reinforcement learning toolkit for compiler optimizations

deep-vector-quantization. VQVAEs, GumbelSoftmaxes and friends

attention-is-all-you-need-pytorch. A PyTorch implementation of the Transformer model in "Attention is All You Need".