wilson1yan Goto Github PK

algorithms

axial-attention

Implementation of Axial attention - attending to multi-dimensional data efficiently

baselines

OpenAI Baselines: high-quality implementations of reinforcement learning algorithms

capsulenet

cater

CATER: A diagnostic dataset for Compositional Actions and TEmporal Reasoning

causal-infogan

clip

Contrastive Language-Image Pretraining

cliport

CLIPort: What and Where Pathways for Robotic Manipulation

collect-habitat

collect-minecraft

contrastive-forward-model

cs231n

cs294-158-lvm

cs294-158-ssl

cs330

deeprl

Modularized Implementation of Deep RL Algorithms in PyTorch

deepspeed

DeepSpeed is a deep learning optimization library that makes distributed training easy, efficient, and effective.

deepul

denoising-diffusion-pytorch

Implementation of Denoising Diffusion Probabilistic Models in PyTorch

dm_control

The DM Control Suite and Package is a tool for developing and testing reinforcement learning agents for the MuJoCo physics engine.

dm_env

A Python interface for reinforcement learning environments

dorm-ex-machina

download-minedojo

dreamerv2

Mastering Atari with Discrete World Models

examples

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

exemplar-vae

Exemplar VAE: Linking Generative Models, Nearest Neighbor Retrieval, and Data Augmentation

fast-transformers

Pytorch library for fast transformer implementations

fastchat

An open platform for training, serving, and evaluating large language models. Release repo for Vicuna and Chatbot Arena.

flaxmodels

Pretrained models for Jax/Flax: StyleGAN2, GPT2, VGG, ResNet, etc.

g-swm

Official PyTorch implementation of "Improving Generative Imagination in Object-Centric World Models"