Tongzhou Wang

dataset-distillation. Open-source code for paper "Dataset Distillation"

align_uniform. Open source code for paper "Understanding Contrastive Representation Learning through Alignment and Uniformity on the Hypersphere" ICML 2020

PyTorch-Reparam-Module. Reparameterize your PyTorch modules

moco_align_uniform. MoCo with Alignment and Uniformity Loss.

poisson_quasimetric_embedding. Open source code for paper "On the Learning and Learnability of Quasimetrics".

pytorch-CycleGAN-and-pix2pix. Image-to-image translation in PyTorch (e.g. horse2zebra, edges2cats, and more)

robodesk. RoboDesk with A Diverse Set of Distractors

distributed_shampoo. For optimization algorithm research and development. Submodule & Py38 ready

CMC_public_fork. pytorch implementation of "Contrastive Multiview Coding", "Momentum Contrast for Unsupervised Visual Representation Learning", and "Unsupervised Feature Learning via Non-Parametric Instance-level Discrimination"

vllm. A high-throughput and memory-efficient inference and serving engine for LLMs

python-billiards. A 2D physics engine for simulating dynamical billiards.

dreamsim. DreamSim: Learning New Dimensions of Human Visual Similarity using Synthetic Data (NeurIPS 2023 Spotlight)

ssnl.github.io. Scratch book

pytorch-generative-model-collections. Collection of generative models in Pytorch version.

compyler. C

rlkit. Collection of reinforcement learning algorithms

vision. Datasets, Transforms and Models specific to Computer Vision

shortest-distance-approx-deep-learning. Implementation of the paper "Shortest Path Distance Approximation using Deep learning Techniques" (under development)

mbold. Model-Based Visual Planning with Self-Supervised Functional Distances (ICLR 2021)

pytorch. Tensors and Dynamic neural networks in Python with strong GPU acceleration

examples. A set of examples around pytorch in Vision, Text, Reinforcement Learning, etc.

LaTeXTools. LaTeX plugin for Sublime Text 2 and 3

Rainbow. Rainbow: Combining Improvements in Deep Reinforcement Learning

dotaPredict. Neural network that predicts results of DotA games